arrow
Return

Improving Neural Network-Based Multi-Label Classification With Pattern Loss Penalties

delete2024-01-01
delete1
delete
OA
AI
W
Worawith Sangkatip
P
Phatthanaphong Chomphuwiset
K
Kaveepoj Bunluewong
S
Sakorn Mekruksavanich
E
Emmanuel Okafor
O
Olarik Surinta *
DOI:10.1109/ACCESS.2024.3386841delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
This research work introduces two novel loss functions, pattern-loss (POL) and label similarity-based instance modeling (LSIM), for improving the performance of multi-label classification using artificial neural network-based techniques. These loss functions incorporate additional optimization constraints based on the distribution of multi-label class patterns and the similarity of data instances. By integrating these patterns during the network training process, the trained model is tuned to align with the existing patterns in the training data. The proposed approach decomposes the loss function into two components: the cross entropy loss and the pattern loss derived from the distribution of class-label patterns. Experimental evaluations were conducted on eight standard datasets, comparing the proposed methods with three existing techniques.The results demonstrate the effectiveness of the proposed approach, with POL and LSIM consistently achieving superior accuracy performance compared to the benchmark methods.
Keywords:
Multi-label classification
label correlation
label-specific features
deep neural network
loss functions

Journal

IEEE Access cover
IEEE Access
IF:
3.6
Papers:
9.8W
Citations:
29.4W

Organization

U
University of Phayao
Scholars:
1.1K
Papers: 872
Citations: 572
M
Mahasarakham University
Scholars:
1.9K
Papers: 1.3K
Citations: 756