arXiv · 2508.13445
ASAP: Unsupervised Post-training with Label Distribution Shift Adaptive Learning Rate
Abstract
In real-world applications, machine learning models face online label shift, where label distributions change over time. Effective adaptation requires careful learning rate selection: too low slows adaptation and too high causes instability. We propose ASAP (Adaptive Shift Aware Post-training), which dynamically adjusts the learning rate by computing the cosine distance between current and previous unlabeled outputs and mapping it within a bounded range. ASAP requires no labels, model ensembles, or past inputs, using only the previous softmax output for fast, lightweight adaptation. Experiments across multiple datasets and shift scenarios show ASAP consistently improves accuracy and efficiency, making it practical for unsupervised model adaptation.
Explore related subjects
Keep this discovery
Heewon Park, Mugon Joe, Miru Kim, Minhae Kwon. 2025-08-19. ASAP: Unsupervised Post-training with Label Distribution Shift Adaptive Learning Rate. https://doi.org/10.1145/3746252.3760830
Cite the original work for its findings. Save a collection to share your selection of sources.