Multi-label active learning with conditional bernoulli mixtures

  • Junyu Chen
  • , Shiliang Sun
  • , Jing Zhao*
  • *Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

4 Scopus citations

Abstract

Multi-label learning is an important machine learning task. In multi-label classification tasks, the label space is larger than the traditional single-label classification, and annotations of multi-label instances are typically more time-consuming or expensive to obtain. Thus, it is necessary to take advantage of active learning to solve this problem. In this paper, we present three active learning methods with the conditional Bernoulli mixture (CBM) model for multi-label classification. The first two methods utilize the least confidence and approximated entropy as the selection criteria to pick the most informative instances, respectively. Particularly, an efficient approximated calculation via dynamic programming is developed to compute the approximated entropy. The third method is based on the cluster information from the CBM, which implicitly takes the advantage of the label correlations. Finally, we demonstrate the effectiveness of the proposed methods through experiments on both synthetic and real-world datasets.

Original languageEnglish
Title of host publicationPRICAI 2018
Subtitle of host publicationTrends in Artificial Intelligence - 15th Pacific Rim International Conference on Artificial Intelligence, Proceedings
EditorsByeong-Ho Kang, Xin Geng
PublisherSpringer Verlag
Pages954-967
Number of pages14
ISBN (Print)9783319973036
DOIs
StatePublished - 2018
Event15th Pacific Rim International Conference on Artificial Intelligence, PRICAI 2018 - Nanjing, China
Duration: 28 Aug 201831 Aug 2018

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume11012 LNAI
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference15th Pacific Rim International Conference on Artificial Intelligence, PRICAI 2018
Country/TerritoryChina
CityNanjing
Period28/08/1831/08/18

Keywords

  • Active learning
  • Machine learning
  • Multi-label classification

Fingerprint

Dive into the research topics of 'Multi-label active learning with conditional bernoulli mixtures'. Together they form a unique fingerprint.

Cite this