Refining the census income prediction through the development of random forest regressor and assessing performance with gradient boosting regression

View at publisher
Authors: Siva Sankar Vattem, S. Loganayagi
Cited by: 0

Abstract

Income prediction using census data is a crucial aspect of socio-economic research, aiding in policy-making and resource allocation. This study aims to refine income prediction models by developing a Random Forest Regressor (RFR) and evaluating its effectiveness against Gradient Boosting Regression (GBR). This research introduces a comparison between ensemble learning techniques through which it evaluates their performance when analysing complex socio-economic datasets with high dimensions. The main goal is to improve predictive model robustness while enhancing accuracy through implementation of feature selection and hyperparameter adjustment with cross-validation processes. The database contains demographic along with socio-economic data that goes through a comprehensive preprocessing phase for imputing missing values while converting categories to numerical forms and normalizing them. The RFR model benefits from grid search and k-fold cross-validation to reach optimization while its performance evaluation compared to GBR model occurs through Mean Absolute Error (MAE), Mean Squared Error (MSE), and R-squared (R 2 ) values. The results indicate RFR achieves superior performance over the models compared to GBR because it manages nonlinear patterns and controls overfitting effectively. Research establishes RFR emerge as the most favourable approach for income modelling in intricate datasets along with GBR which demonstrates strong capability. The research outcomes advance machine learning applications for socio-economic modelling which serves as fundamental knowledge for data-driven policies development.

Fields of Study & Subjects

Data Mining and Machine Learning Applications Artificial Intelligence in Healthcare Machine Learning and Data Classification

Publication details

Journal / source
CRC Press eBooks
Pages
303–309
Publication date
5 March 2026
Language
EN
Document type
Book chapter