Probabilistic multi-dimensional classification with incomplete data at the prediction time

Thu Ha DO, Vu-Linh Nguyen, Yves Grandvalet

TL;DR: We present a new probabilistic multi-dimensional classification approach with efficient learning and prediction algorithms that address scalability and robustness issues.

Abstract: Multi-dimensional classification (MDC) extends multi-class and multi-label learning by predicting several class variables per instance. We revisit probabilistic MDC methods with mixed features (discrete and continuous), focusing on their strengths and limits for handling incomplete data at prediction time. We present theoretical results leading to a new probabilistic approach with efficient learning and prediction algorithms that address scalability and robustness issues. Experiments demonstrate its benefits in different missingness scenarios.

Code Dataset Promise: Yes

Code Dataset Url: https://github.com/telagment/HMDC_tabular

Format Confirmation: I agree that I have read and followed the formatting instructions for the camera ready version.

Submission Number: 15