LCBM : A Multi-View Probabilistic Model for Multi-Label Classification

Multi-label classification is an important research topic in machine learning, for which exploiting label dependencies is an effective modeling principle. Recently, probabilistic models have shown great potential in discovering dependencies among labels. In this paper, motivated by the recent success of multi-view learning to improve the generalization performance, we propose a novel multi-view probabilistic model named latent conditional Bernoulli mixture (LCBM) for multi-label classification. LCBM is a generative model taking features from different views as inputs, and conditional on the latent subspace shared by the views a Bernoulli mixture model is adopted to build label dependencies. Inside each component of the mixture, the labels have a weak correlation which facilitates computational convenience. The mean field variational inference framework is used to carry out approximate posterior inference in the probabilistic model, where we propose a Gaussian mixture variational autoencoder (GMVAE) for effective posterior approximation. We further develop a scalable stochastic training algorithm for efficiently optimizing the model parameters and variational parameters, and derive an efficient prediction procedure based on greedy search. Experimental results on multiple benchmark datasets show that our approach outperforms other state-of-the-art methods under various metrics.

Medienart:

E-Artikel

Erscheinungsjahr:

2021

Erschienen:

2021

Enthalten in:

Zur Gesamtaufnahme - volume:43

Enthalten in:

IEEE transactions on pattern analysis and machine intelligence - 43(2021), 8 vom: 20. Aug., Seite 2682-2696

Sprache:

Englisch

Beteiligte Personen:

Sun, Shiliang [VerfasserIn]
Zong, Daoming [VerfasserIn]

Links:

Volltext

Themen:

Journal Article

Anmerkungen:

Date Revised 02.07.2021

published: Print-Electronic

Citation Status PubMed-not-MEDLINE

doi:

10.1109/TPAMI.2020.2974203

funding:

Förderinstitution / Projekttitel:

PPN (Katalog-ID):

NLM306723506