Language-Guided Reinforcement Learning for Hard Attention in Few-Shot Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Nikpour, Bahareh, Armanfard, Narges
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909585093165056
author Nikpour, Bahareh
Armanfard, Narges
author_facet Nikpour, Bahareh
Armanfard, Narges
contents Attention mechanisms have demonstrated significant potential in enhancing learning models by identifying key portions of input data, particularly in scenarios with limited training samples. Inspired by human perception, we propose that focusing on essential data segments, rather than the entire dataset, can improve the accuracy and reliability of the learning models. However, identifying these critical data segments, or "hard attention finding," is challenging, especially in few-shot learning, due to the scarcity of training data and the complexity of model parameters. To address this, we introduce LaHA, a novel framework that leverages language-guided deep reinforcement learning to identify and utilize informative data regions, thereby improving both interpretability and performance. Extensive experiments on benchmark datasets validate the effectiveness of LaHA.
format Preprint
id arxiv_https___arxiv_org_abs_2310_07800
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Language-Guided Reinforcement Learning for Hard Attention in Few-Shot Learning
Nikpour, Bahareh
Armanfard, Narges
Artificial Intelligence
Machine Learning
Attention mechanisms have demonstrated significant potential in enhancing learning models by identifying key portions of input data, particularly in scenarios with limited training samples. Inspired by human perception, we propose that focusing on essential data segments, rather than the entire dataset, can improve the accuracy and reliability of the learning models. However, identifying these critical data segments, or "hard attention finding," is challenging, especially in few-shot learning, due to the scarcity of training data and the complexity of model parameters. To address this, we introduce LaHA, a novel framework that leverages language-guided deep reinforcement learning to identify and utilize informative data regions, thereby improving both interpretability and performance. Extensive experiments on benchmark datasets validate the effectiveness of LaHA.
title Language-Guided Reinforcement Learning for Hard Attention in Few-Shot Learning
topic Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2310.07800