LLM Enhanced Machine Learning Estimators for Classification
Pre-trained large language models (LLM) have emerged as a powerful tool for simulating various scenarios and generating informative output given specific instructions and multimodal input. In this work, we analyze the specific use of LLM to enhance a classical supervised machine learning method for classification problems. We propose a few approaches to integrate LLM into a classical machine learning estimator to further enhance the prediction performance. We examine the performance of the proposed approaches through both standard supervised learning binary classification tasks, and a transfer learning task where the test data observe distribution changes compared to the training data. Numerical experiments using four publicly available datasets are conducted and suggest that using LLM to enhance classical machine learning estimators can provide significant improvement on prediction performance.
Authors
- Chu Wang (ORCID: https://orcid.org/0000-0002-7889-6981)
- Zeyu Zheng
- Yingfei Wang (ORCID: https://orcid.org/0000-0002-3634-3617)
- Yuhang Wu (ORCID: https://orcid.org/0000-0002-8681-1538)
Institutions
- Amazon (United States) (US)
- University of Washington (US)
- University of California, Berkeley (US)
Publication Details
- Published
- 2024-12-15
- DOI
- https://doi.org/10.1109/wsc63780.2024.10838779
- Citations
- 1
- Primary Topic
- Neural Networks and Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.35