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.
Til et andet sprog
fra kildeindhold
arxiv.org
Vigtigste indsigter udtrukket fra
by Ao Zhou,Bin ... kl. arxiv.org 03-28-2024
https://arxiv.org/pdf/2403.18192.pdfDybere Forespørgsler