Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. Addressing Label Shift In Distributed Learning Via Entropy Regularization
 
conference paper not in proceedings

Addressing Label Shift In Distributed Learning Via Entropy Regularization

Wu, Zhiyuan
•
Choi, Changkyu
•
Cao, Xiangcheng
Show more
April 2025
The Thirteenth International Conference on Learning Representations

We address the challenge of minimizing true risk in multi-node distributed learning. 1 These systems are frequently exposed to both inter-node and intra-node label shifts, which present a critical obstacle to effectively optimizing model performance while ensuring that data remains confined to each node. To tackle this, we propose the Versatile Robust Label Shift (VRLS) method, which enhances the maximum likelihood estimation of the test-to-train label importance ratio. VRLS incorporates Shannon entropy-based regularization and adjusts the importance ratio during training to better handle label shifts at the test time. In multi-node learning environments, VRLS further extends its capabilities by learning and adapting importance ratios across nodes, effectively mitigating label shifts and improving overall model performance. Experiments conducted on MNIST, Fashion MNIST, and CIFAR-10 demonstrate the effectiveness of VRLS, outperforming baselines by up to 20% in imbalanced settings. These results highlight the significant improvements VRLS offers in addressing label shifts. Our theoretical analysis further supports this by establishing high-probability bounds on estimation errors. The code is available at https://github.com/zhiyuan-11/VRLS_main/tree/main.

  • Files
  • Details
  • Metrics
Type
conference paper not in proceedings
Author(s)
Wu, Zhiyuan
Choi, Changkyu
Cao, Xiangcheng
Cevher, Volkan  orcid-logo

EPFL

Ramezani-Kebrya, Ali
Date Issued

2025-04

Subjects

ML-AI

Written at

EPFL

EPFL units
LIONS  
Event nameEvent acronymEvent placeEvent date
The Thirteenth International Conference on Learning Representations

ICLR

Singapore

2025-04-24-2025-04-28

Available on Infoscience
May 5, 2025
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/249795
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés