Harnessing Large Language Models as Post-hoc Correctors

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhong, Zhiqiang, Zhou, Kuangyu, Mottin, Davide
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916299566743552
author Zhong, Zhiqiang
Zhou, Kuangyu
Mottin, Davide
author_facet Zhong, Zhiqiang
Zhou, Kuangyu
Mottin, Davide
contents As Machine Learning (ML) models grow in size and demand higher-quality training data, the expenses associated with re-training and fine-tuning these models are escalating rapidly. Inspired by recent impressive achievements of Large Language Models (LLMs) in different fields, this paper delves into the question: can LLMs efficiently improve an ML's performance at a minimal cost? We show that, through our proposed training-free framework LlmCorr, an LLM can work as a post-hoc corrector to propose corrections for the predictions of an arbitrary ML model. In particular, we form a contextual knowledge database by incorporating the dataset's label information and the ML model's predictions on the validation dataset. Leveraging the in-context learning capability of LLMs, we ask the LLM to summarise the instances in which the ML model makes mistakes and the correlation between primary predictions and true labels. Following this, the LLM can transfer its acquired knowledge to suggest corrections for the ML model's predictions. Our experimental results on text analysis and the challenging molecular predictions show that \model improves the performance of a number of models by up to 39%.
format Preprint
id arxiv_https___arxiv_org_abs_2402_13414
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Harnessing Large Language Models as Post-hoc Correctors
Zhong, Zhiqiang
Zhou, Kuangyu
Mottin, Davide
Machine Learning
Computation and Language
As Machine Learning (ML) models grow in size and demand higher-quality training data, the expenses associated with re-training and fine-tuning these models are escalating rapidly. Inspired by recent impressive achievements of Large Language Models (LLMs) in different fields, this paper delves into the question: can LLMs efficiently improve an ML's performance at a minimal cost? We show that, through our proposed training-free framework LlmCorr, an LLM can work as a post-hoc corrector to propose corrections for the predictions of an arbitrary ML model. In particular, we form a contextual knowledge database by incorporating the dataset's label information and the ML model's predictions on the validation dataset. Leveraging the in-context learning capability of LLMs, we ask the LLM to summarise the instances in which the ML model makes mistakes and the correlation between primary predictions and true labels. Following this, the LLM can transfer its acquired knowledge to suggest corrections for the ML model's predictions. Our experimental results on text analysis and the challenging molecular predictions show that \model improves the performance of a number of models by up to 39%.
title Harnessing Large Language Models as Post-hoc Correctors
topic Machine Learning
Computation and Language
url https://arxiv.org/abs/2402.13414