LLMDR: Large language model driven framework for missing data recovery in mixed data under low resource regime
About
The missing data problem is one of the important issues to address for achieving data quality. While imputation-based methods are designed to achieve data completeness, their efficacy is observed to be diminishing as and when there is increasing in the missingness percentage. Further, extant approaches often struggle to handle mixed-type datasets, typically supporting either numerical and/or categorical data. In this work, we propose LLMDR, automatic data recovery framework which operates in two stage approach, wherein the Stage-I: DBSCAN clustering algorithm is employed to select the most representative samples and in the Stage-II: Multi-LLMs are employed for data recovery considering the local and global representative samples; Later, this framework invokes the consensus algorithm for recommending a more accurate value based on other LLMs of local and global effective samples. Experimental results demonstrate that proposed framework works effectively on various mixed datasets in terms of Accuracy, KS-Statistic, SMAPE, and MSE. Further, we have also shown the advantage of the consensus mechanism for final recommendation in mixed-type data.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Value Recommendation for Missing Data | Buy | Accuracy76.6667 | 9 | |
| Missing Value Imputation | Phone dataset 10% Missing Rate | Accuracy73.3333 | 6 | |
| Missing Value Imputation | Phone dataset 20% Missing Rate | Accuracy66.6667 | 6 | |
| Missing Value Imputation | Phone dataset 30% Missing Rate | Accuracy (%)65.5556 | 6 | |
| Data Imputation | Restaurant 10% Missing Rate | Accuracy61.5385 | 5 | |
| Data Imputation | Restaurant 20% Missing Rate | Accuracy65.3846 | 5 | |
| Data Imputation | Restaurant 30% Missing Rate | Accuracy (%)61.5385 | 5 |