Die Studie untersucht das Problem der Klassenungleichgewichte in der Multi-Label-Klassifizierung und deren Auswirkungen auf das Training von Deep-Learning-Modellen. Es wird ein neuartiger Multi-Label-Batch-Auswahlalgorithmus vorgestellt, der sich auf Klassenungleichgewichte und schwierige Samples konzentriert.
Der Algorithmus wählt Batches adaptiv aus, indem er die Binary Cross Entropy-Verluste mit einer Kombination aus globalen und lokalen Ungleichgewichtsgewichten gewichtet. Dadurch erhalten herausfordernde Samples, die oft mit Minderheitenlabels assoziiert sind, eine höhere Priorität. Außerdem wird die Ranking-Strategie durch Quantisierungsfaktoren verfeinert, um das Problem der Glättung anzugehen.
Es wird auch eine Variante des adaptiven Batch-Auswahlverfahrens eingeführt, die Labelkorrelationen berücksichtigt. Umfangreiche Experimente zeigen, dass der Ansatz zu einer besseren Konvergenz und Leistung im Vergleich zu den in bestehenden Modellen verwendeten zufälligen Batch-Auswahlmethoden führt.
다른 언어로
소스 콘텐츠 기반
arxiv.org
더 깊은 질문