Loss Design and Architecture Selection for Long-Tailed Multi-Label Chest X-Ray Classification
Organizations: Department of Computer Science and Engineering, Rajiv Gandhi University of Knowledge Technologies, Nuzvid, India
Abstract
Long-tailed class distributions pose a significant challenge for multi-label chest X-ray (CXR) classification, where rare but clinically important findings are severely underrepresented. In this work, we present a systematic empirical evaluation of loss functions, CNN backbone architectures and post-training strategies on the CXR-LT 2026 benchmark, comprising approximately 143K images with 30 disease labels from PadChest. Our experiments demonstrate that LDAM with deferred re-weighting (LDAM-DRW) consistently outperforms standard BCE and asymmetric losses for rare class recognition. Amongst the architectures evaluated, ConvNeXt-Large achieves the best single-model performance with 0.5220 mAP and 0.3765 F1 on our development set, whilst classifier re-training and test-time augmentation further improve ranking metrics. On the official test leaderboard, our submission achieved 0.3950 mAP, ranking 5th amongst all 68 participating teams with total of 1528 submissions. We provide a candid analysis of the development-to-test performance gap and discuss practical insights for handling class imbalance in clinical imaging settings. Code is available at https://github.com/Nikhil-Rao20/Long_Tail.
Figures & tables
| Model | Loss | Post-training | AP | AUC | F1 | ECE | |
| ResNet-50 | BCE | – | |||||
| ResNet-50 | Asymmetric | – | |||||
| ResNet-50 | LDAM+DRW | – | |||||
| ResNet-50 | LDAM+DRW | cRT | |||||
| ResNet-50 | LDAM+DRW | cRT + TTA | |||||
| ResNet-101 | LDAM+DRW | – |
| Rank | Team | Affiliation | mAP | AUC | F1 |
|---|---|---|---|---|---|
| 1 | CVMAIL x MIHL | Vietnam National University, Vietnam | 0.5854 | 0.9259 | 0.3518 |
| 2 | Cool Peace | KAIST Graduate School of AI, Korea | 0.4827 | 0.9186 | 0.3162 |
| 3 | VIU | Vietnam National University, Vietnam | 0.4599 | 0.8827 | 0.4504 |
| 4 | Bibimbap-Bueno | Case Western Reserve University, USA | 0.4297 | 0.8753 | 0.2482 |
| 5 | Nikhil Rao Sulake | RGUKT Nuzvid, India | 0.3950 | 0.8591 | 0.0945 |
| 6 | UGIVIA team | Universitat de les Illes Balears, Spain | 0.2362 | 0.7756 | 0.2353 |