Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

When Does Aggregating Multiple Skills with Multi-Task Learning Work? A Case Study in Financial NLP

About

Multi-task learning (MTL) aims at achieving a better model by leveraging data and knowledge from multiple tasks. However, MTL does not always work -- sometimes negative transfer occurs between tasks, especially when aggregating loosely related skills, leaving it an open question when MTL works. Previous studies show that MTL performance can be improved by algorithmic tricks. However, what tasks and skills should be included is less well explored. In this work, we conduct a case study in Financial NLP where multiple datasets exist for skills relevant to the domain, such as numeric reasoning and sentiment analysis. Due to the task difficulty and data scarcity in the Financial NLP domain, we explore when aggregating such diverse skills from multiple datasets with MTL can work. Our findings suggest that the key to MTL success lies in skill diversity, relatedness between tasks, and choice of aggregation size and shared capacity. Specifically, MTL works well when tasks are diverse but related, and when the size of the task aggregation and the shared capacity of the model are balanced to avoid overwhelming certain tasks.

Jingwei Ni, Zhijing Jin, Qian Wang, Mrinmaya Sachan, Markus Leippold• 2023

Related benchmarks

TaskDatasetResultRank
Financial Natural Language ProcessingFinDATA
TSA0.2159
10
Sentiment ClassificationMalo SC 2013
Accuracy86.61
3
Target-Based Sentiment AnalysisTSA 2017
RMSE0.2054
3
Showing 3 of 3 rows

Other info

Code

Follow for update