Learning Label Distribution with Dirichlet Process Mixture Model

Read full text
Authors: Association for Artificial Intelligence 2026, Xiuyi Jia, Weiwei Li, Yunan Lu, Minglong Wang
Cited by: 0

Abstract

Label Distribution Learning (LDL) is an effective machine learning paradigm for addressing label ambiguity, where each sample is annotated with a distribution that conveys rich semantic information. However, during the actual annotation process of label distributions, annotators often exhibit divergent labeling preferences for the same sample. Most existing LDL methods overlook this heterogeneity, assuming that the observed label distribution originates from a single labeling pattern. Such an assumption limits their capacity to manage inter-annotator disagreement and constrains the generalization of the resulting models. To address this issue, we propose, for the first time, a Dirichlet process mixture model (DPMM)-based framework for LDL. This framework leverages nonparametric Bayesian methods to adaptively uncover diverse latent labeling patterns from the data and to accurately model annotator heterogeneity. Specifically, the ground-truth label distribution of each sample is modeled as a weighted mixture of multiple latent components, where a feature-conditioned gating mechanism adaptively controls the contribution of each component. Experimental results demonstrate that the proposed model consistently achieves competitive performance on several widely-used benchmark datasets.

Fields of Study & Subjects

Machine Learning and Data Classification Bayesian Methods and Mixture Models Domain Adaptation and Few-Shot Learning

Publication details

Journal / source
Underline Science Inc.
Publication date
7 January 2026
Language
EN