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.
A otro idioma
del contenido fuente
arxiv.org
Ideas clave extraídas de
by Ao Zhou,Bin ... a las arxiv.org 03-28-2024
https://arxiv.org/pdf/2403.18192.pdfConsultas más profundas