GAIS: A Novel Approach to Instance Selection with Graph Attention Networks

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Rustamov, Zahiriddin, Zaitouny, Ayham, Damseh, Rafat, Zaki, Nazar
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910764661473280
author Rustamov, Zahiriddin
Zaitouny, Ayham
Damseh, Rafat
Zaki, Nazar
author_facet Rustamov, Zahiriddin
Zaitouny, Ayham
Damseh, Rafat
Zaki, Nazar
contents Instance selection (IS) is a crucial technique in machine learning that aims to reduce dataset size while maintaining model performance. This paper introduces a novel method called Graph Attention-based Instance Selection (GAIS), which leverages Graph Attention Networks (GATs) to identify the most informative instances in a dataset. GAIS represents the data as a graph and uses GATs to learn node representations, enabling it to capture complex relationships between instances. The method processes data in chunks, applies random masking and similarity thresholding during graph construction, and selects instances based on confidence scores from the trained GAT model. Experiments on 13 diverse datasets demonstrate that GAIS consistently outperforms traditional IS methods in terms of effectiveness, achieving high reduction rates (average 96\%) while maintaining or improving model performance. Although GAIS exhibits slightly higher computational costs, its superior performance in maintaining accuracy with significantly reduced training data makes it a promising approach for graph-based data selection.
format Preprint
id arxiv_https___arxiv_org_abs_2412_19201
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle GAIS: A Novel Approach to Instance Selection with Graph Attention Networks
Rustamov, Zahiriddin
Zaitouny, Ayham
Damseh, Rafat
Zaki, Nazar
Machine Learning
Instance selection (IS) is a crucial technique in machine learning that aims to reduce dataset size while maintaining model performance. This paper introduces a novel method called Graph Attention-based Instance Selection (GAIS), which leverages Graph Attention Networks (GATs) to identify the most informative instances in a dataset. GAIS represents the data as a graph and uses GATs to learn node representations, enabling it to capture complex relationships between instances. The method processes data in chunks, applies random masking and similarity thresholding during graph construction, and selects instances based on confidence scores from the trained GAT model. Experiments on 13 diverse datasets demonstrate that GAIS consistently outperforms traditional IS methods in terms of effectiveness, achieving high reduction rates (average 96\%) while maintaining or improving model performance. Although GAIS exhibits slightly higher computational costs, its superior performance in maintaining accuracy with significantly reduced training data makes it a promising approach for graph-based data selection.
title GAIS: A Novel Approach to Instance Selection with Graph Attention Networks
topic Machine Learning
url https://arxiv.org/abs/2412.19201