PiCO: Contrastive Label Disambiguation for Partial Label Learning
  Partial label learning (PLL) is an important problem that allows each
training example to be labeled with a coarse candidate set, which well suits
many real-world data annotation scenarios with label ambiguity. Despite the
promise, the performance of PLL often lags behind the supervised counterpart.
In this work, we bridge the gap by addressing two key research challenges in
PLL -- representation learning and label disambiguation -- in one coherent
framework. Specifically, our proposed framework PiCO consists of a contrastive
learning module along with a novel class prototype-based label disambiguation
algorithm. PiCO produces closely aligned representations for examples from the
same classes and facilitates label disambiguation. Theoretically, we show that
these two components are mutually beneficial, and can be rigorously justified
from an expectation-maximization (EM) algorithm perspective. Extensive
experiments demonstrate that PiCO significantly outperforms the current
state-of-the-art approaches in PLL and even achieves comparable results to
fully supervised learning. Code and data available:
https://github.com/hbzju/PiCO.
