2412.11455

Total: 1

#1 Towards Better Multi-task Learning: A Framework for Optimizing Dataset Combinations in Large Language Models [PDF1] [Copy] [Kimi] [REL]

Authors: Zaifu Zhan, Rui Zhang

To efficiently select optimal dataset combinations for enhancing multi-task learning (MTL) performance in large language models, we proposed a novel framework that leverages a neural network to predict the best dataset combinations. The framework iteratively refines the selection, greatly improving efficiency, while being model-, dataset-, and domain-independent. Through experiments on 12 biomedical datasets across four tasks - named entity recognition, relation extraction, event extraction, and text classification-we demonstrate that our approach effectively identifies better combinations, even for tasks that may seem unpromising from a human perspective. This verifies that our framework provides a promising solution for maximizing MTL potential.

Subjects: Computation and Language , Artificial Intelligence

Publish: 2024-12-16 05:20:18 UTC