Exploring the Effectiveness and Consistency of Task Selection in Intermediate-Task Transfer Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lin, Pin-Jie, Zhang, Miaoran, Mosbach, Marius, Klakow, Dietrich
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914882577760256
author Lin, Pin-Jie
Zhang, Miaoran
Mosbach, Marius
Klakow, Dietrich
author_facet Lin, Pin-Jie
Zhang, Miaoran
Mosbach, Marius
Klakow, Dietrich
contents Identifying beneficial tasks to transfer from is a critical step toward successful intermediate-task transfer learning. In this work, we experiment with 130 source-target task combinations and demonstrate that the transfer performance exhibits severe variance across different source tasks and training seeds, highlighting the crucial role of intermediate-task selection in a broader context. We compare four representative task selection methods in a unified setup, focusing on their effectiveness and consistency. Compared to embedding-free methods and text embeddings, task embeddings constructed from fine-tuned weights can better estimate task transferability by improving task prediction scores from 2.59% to 3.96%. Despite their strong performance, we observe that the task embeddings do not consistently demonstrate superiority for tasks requiring reasoning abilities. Furthermore, we introduce a novel method that measures pairwise token similarity using maximum inner product search, leading to the highest performance in task prediction. Our findings suggest that token-wise similarity is better predictive for predicting transferability compared to averaging weights.
format Preprint
id arxiv_https___arxiv_org_abs_2407_16245
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Exploring the Effectiveness and Consistency of Task Selection in Intermediate-Task Transfer Learning
Lin, Pin-Jie
Zhang, Miaoran
Mosbach, Marius
Klakow, Dietrich
Computation and Language
Identifying beneficial tasks to transfer from is a critical step toward successful intermediate-task transfer learning. In this work, we experiment with 130 source-target task combinations and demonstrate that the transfer performance exhibits severe variance across different source tasks and training seeds, highlighting the crucial role of intermediate-task selection in a broader context. We compare four representative task selection methods in a unified setup, focusing on their effectiveness and consistency. Compared to embedding-free methods and text embeddings, task embeddings constructed from fine-tuned weights can better estimate task transferability by improving task prediction scores from 2.59% to 3.96%. Despite their strong performance, we observe that the task embeddings do not consistently demonstrate superiority for tasks requiring reasoning abilities. Furthermore, we introduce a novel method that measures pairwise token similarity using maximum inner product search, leading to the highest performance in task prediction. Our findings suggest that token-wise similarity is better predictive for predicting transferability compared to averaging weights.
title Exploring the Effectiveness and Consistency of Task Selection in Intermediate-Task Transfer Learning
topic Computation and Language
url https://arxiv.org/abs/2407.16245