A Pytorch Implementation of Balanced Sharpness-Aware Minimization for Imbalanced Regression (ICCV 2025)
conda env create -f environment.yml
We provide the code for AgeDB-DIR and IMDB-WIKI-DIR dataset. Please go into the subfolders for more information.
We provide the checkpoints at Google Drive.
@InProceedings{Liu_2025_ICCV,
author = {Liu, Yahao and Wang, Qin and Duan, Lixin and Li, Wen},
title = {Balanced Sharpness-Aware Minimization for Imbalanced Regression},
booktitle = {The IEEE International Conference on Computer Vision (ICCV)},
year = {2025}
}
Some codes are adapted from imbalanced-regression and Rank-N-Contrast. We thank them for their excellent projects.