Potential Energy based Mixture Model for Noisy Label Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Zijia, Yang, Wenbin, Liu, Zhisong, Jia, Zhen
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910431911608320
author Wang, Zijia
Yang, Wenbin
Liu, Zhisong
Jia, Zhen
author_facet Wang, Zijia
Yang, Wenbin
Liu, Zhisong
Jia, Zhen
contents Training deep neural networks (DNNs) from noisy labels is an important and challenging task. However, most existing approaches focus on the corrupted labels and ignore the importance of inherent data structure. To bridge the gap between noisy labels and data, inspired by the concept of potential energy in physics, we propose a novel Potential Energy based Mixture Model (PEMM) for noise-labels learning. We innovate a distance-based classifier with the potential energy regularization on its class centers. Embedding our proposed classifier with existing deep learning backbones, we can have robust networks with better feature representations. They can preserve intrinsic structures from the data, resulting in a superior noisy tolerance. We conducted extensive experiments to analyze the efficiency of our proposed model on several real-world datasets. Quantitative results show that it can achieve state-of-the-art performance.
format Preprint
id arxiv_https___arxiv_org_abs_2405_01186
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Potential Energy based Mixture Model for Noisy Label Learning
Wang, Zijia
Yang, Wenbin
Liu, Zhisong
Jia, Zhen
Machine Learning
Artificial Intelligence
Training deep neural networks (DNNs) from noisy labels is an important and challenging task. However, most existing approaches focus on the corrupted labels and ignore the importance of inherent data structure. To bridge the gap between noisy labels and data, inspired by the concept of potential energy in physics, we propose a novel Potential Energy based Mixture Model (PEMM) for noise-labels learning. We innovate a distance-based classifier with the potential energy regularization on its class centers. Embedding our proposed classifier with existing deep learning backbones, we can have robust networks with better feature representations. They can preserve intrinsic structures from the data, resulting in a superior noisy tolerance. We conducted extensive experiments to analyze the efficiency of our proposed model on several real-world datasets. Quantitative results show that it can achieve state-of-the-art performance.
title Potential Energy based Mixture Model for Noisy Label Learning
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2405.01186