English

Data Augmentation via Diffusion Model to Enhance AI Fairness

Machine Learning 2024-10-22 v1 Artificial Intelligence Computers and Society

Abstract

AI fairness seeks to improve the transparency and explainability of AI systems by ensuring that their outcomes genuinely reflect the best interests of users. Data augmentation, which involves generating synthetic data from existing datasets, has gained significant attention as a solution to data scarcity. In particular, diffusion models have become a powerful technique for generating synthetic data, especially in fields like computer vision. This paper explores the potential of diffusion models to generate synthetic tabular data to improve AI fairness. The Tabular Denoising Diffusion Probabilistic Model (Tab-DDPM), a diffusion model adaptable to any tabular dataset and capable of handling various feature types, was utilized with different amounts of generated data for data augmentation. Additionally, reweighting samples from AIF360 was employed to further enhance AI fairness. Five traditional machine learning models-Decision Tree (DT), Gaussian Naive Bayes (GNB), K-Nearest Neighbors (KNN), Logistic Regression (LR), and Random Forest (RF)-were used to validate the proposed approach. Experimental results demonstrate that the synthetic data generated by Tab-DDPM improves fairness in binary classification.

Keywords

Cite

@article{arxiv.2410.15470,
  title  = {Data Augmentation via Diffusion Model to Enhance AI Fairness},
  author = {Christina Hastings Blow and Lijun Qian and Camille Gibson and Pamela Obiomon and Xishuang Dong},
  journal= {arXiv preprint arXiv:2410.15470},
  year   = {2024}
}

Comments

arXiv admin note: text overlap with arXiv:2312.12560

R2 v1 2026-06-28T19:28:50.872Z