Explainable Machine Learning Framework for Income Prediction with Class Imbalance Optimization

Income prediction from demographic data remains challenging due to inherent class imbalance and the black box nature of modern machine learning algorithms.This study develops a comprehensive explainable AI framework to predict income levels using the Adult Income dataset while addressing the critical 76/24 class distribution skew.The research implements and compares four state of the art algorithms (XGBoost, LightGBM, RandomForest, and CatBoost) enhanced with SMOTE balancing and optimal threshold selection.Through systematic application of SHAP, LIME, and permutation importance methods, the framework provides transparent model interpretability.Results demonstrate that LightGBM achieves the best performance with 72.91% F1 score and 82.23% balanced accuracy after threshold optimization, representing a significant improvement over baseline models.The XAI analysis reveals marital status and capital gains as dominant predictive features, with strong consensus across explainability methods.Learning curve analysis confirms model convergence at approximately 35,000 samples with minimal overfitting gaps below 3%.The framework's novelty lies in combining multiple explainability techniques with systematic threshold optimization for imbalanced data.These findings have important implications for fair and transparent automated decision making in financial services, lending, and human resource applications where understanding model reasoning is crucial.

Institutions

Publication Details

Journal
IJARCCE
Published
2025-08-14
DOI
https://doi.org/10.17148/ijarcce.2025.14801
Primary Topic
Financial Distress and Bankruptcy Prediction
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Explainable Machine Learning Framework for Income Prediction with Class Imbalance Optimization

IJARCCE
Financial Distress and Bankruptcy Prediction
article

Explainable Machine Learning Framework for Income Prediction with Class Imbalance Optimization

article en

Abstract

Income prediction from demographic data remains challenging due to inherent class imbalance and the black box nature of modern machine learning algorithms.This study develops a comprehensive explainable AI framework to predict income levels using the Adult Income dataset while addressing the critical 76/24 class distribution skew.The research implements and compares four state of the art algorithms (XGBoost, LightGBM, RandomForest, and CatBoost) enhanced with SMOTE balancing and optimal threshold selection.Through systematic application of SHAP, LIME, and permutation importance methods, the framework provides transparent model interpretability.Results demonstrate that LightGBM achieves the best performance with 72.91% F1 score and 82.23% balanced accuracy after threshold optimization, representing a significant improvement over baseline models.The XAI analysis reveals marital status and capital gains as dominant predictive features, with strong consensus across explainability methods.Learning curve analysis confirms model convergence at approximately 35,000 samples with minimal overfitting gaps below 3%.The framework's novelty lies in combining multiple explainability techniques with systematic threshold optimization for imbalanced data.These findings have important implications for fair and transparent automated decision making in financial services, lending, and human resource applications where understanding model reasoning is crucial.

IJARCCEVol. 14(8)
Federal University Otuoke (NG)
Openalex Percentile: Top 68%
Financial Distress and Bankruptcy Prediction
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.