Paper Details

Abstract

Chest X-ray classification suffers from severe class imbalance where gradient updates bias toward majority classes, causing feature drift and poor performance on rare but critical pathologies. We propose a Momentum-Anchored Multi-Scale Fusion Network that uses exponential moving averages (EMA) as a temporal anchoring mechanism to stabilize feature representations under long-tailed distributions. Our approach applies selective momentum updates to the final expansion block of an EfficientNet backbone, creating a slowly-evolving reference branch that resists gradient-induced drift while preserving discriminative patterns for minority classes. Combined with multi-scale spatial fusion (1×1, 3×3, 5×5 convolutions), this anchoring strategy maintains representational sta bility throughout training. On ChestX-ray14, our method achieves 0.8682 average AUC, outperforming state-of-the-art approaches and showing particular improvements on rare pathologies like Hernia (0.9470) and Pneumonia (0.8165). The results demonstrate that momentum anchoring effectively counters feature instability in long-tailed medical image classification.

Keywords
Memory-augmented neural networks chest X-ray class imbalance deep learning medical imaging
Contact Information
Ngu Cong Viet Huynh (Corresponding Author)
FPT University, Vietnam
0968683264

All Authors (3)

Duy Khuong Hoang

Affiliation: FPT University

Country: Vietnam

Email: duyhkse184883@fpt.edu.vn

Phone: 0914145907

Duy Huu Nguyen

Affiliation: FPT University

Country: Vietnam

Email: duynhse183995@fpt.edu.vn

Phone: 0839973335

Ngu Cong Viet Huynh C

Affiliation: FPT University

Country: Vietnam

Email: nguhcv@fe.edu.vn

Phone: 0968683264