Computer Science > Machine Learning
arXiv:2405.18296 (cs)
[Submitted on 28 May 2024 (v1), last revised 22 Dec 2024 (this version, v2)]
Title:Bias in Motion: Theoretical Insights into the Dynamics of Bias in SGD Training
View a PDF of the paper titled Bias in Motion: Theoretical Insights into the Dynamics of Bias in SGD Training, by Anchit Jain and 3 other authors
View PDFHTML (experimental)Abstract:Machine learning systems often acquire biases by leveraging undesired features in the data, impacting accuracy variably across different sub-populations. Current understanding of bias formation mostly focuses on the initial and final stages of learning, leaving a gap in knowledge regarding the transient dynamics. To address this gap, this paper explores the evolution of bias in a teacher-student setup modeling different data sub-populations with a Gaussian-mixture model. We provide an analytical description of the stochastic gradient descent dynamics of a linear classifier in this setting, which we prove to be exact in high dimension. Notably, our analysis reveals how different properties of sub-populations influence bias at different timescales, showing a shifting preference of the classifier during training. Applying our findings to fairness and robustness, we delineate how and when heterogeneous data and spurious features can generate and amplify bias. We empirically validate our results in more complex scenarios by training deeper networks on synthetic and real datasets, including CIFAR10, MNIST, and CelebA.
| Comments: | NeurIPS 2024 |
| Subjects: | Machine Learning (cs.LG); Disordered Systems and Neural Networks (cond-mat.dis-nn); Machine Learning (stat.ML) |
| Cite as: | arXiv:2405.18296 [cs.LG] |
| (orarXiv:2405.18296v2 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2405.18296 arXiv-issued DOI via DataCite |
Submission history
From: Anchit Jain [view email][v1] Tue, 28 May 2024 15:50:10 UTC (1,133 KB)
[v2] Sun, 22 Dec 2024 19:59:00 UTC (1,745 KB)
Full-text links:
Access Paper:
- View PDF
- HTML (experimental)
- TeX Source
- Other Formats
View a PDF of the paper titled Bias in Motion: Theoretical Insights into the Dynamics of Bias in SGD Training, by Anchit Jain and 3 other authors
Current browse context:
cs.LG
Change to browse by:
References & Citations
Bookmark
Bibliographic and Citation Tools
Bibliographic Explorer(What is the Explorer?)
Connected Papers(What is Connected Papers?)
Litmaps(What is Litmaps?)
scite Smart Citations(What are Smart Citations?)
Code, Data and Media Associated with this Article
alphaXiv(What is alphaXiv?)
CatalyzeX Code Finder for Papers(What is CatalyzeX?)
DagsHub(What is DagsHub?)
Gotit.pub(What is GotitPub?)
Hugging Face(What is Huggingface?)
Papers with Code(What is Papers with Code?)
ScienceCast(What is ScienceCast?)
Demos
Recommenders and Search Tools
Influence Flower(What are Influence Flowers?)
CORE Recommender(What is CORE?)
IArxiv Recommender(What is IArxiv?)
arXivLabs: experimental projects with community collaborators
arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.
Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.
Have an idea for a project that will add value for arXiv's community?Learn more about arXivLabs.