Saved in:
Bibliographic Details
Main Authors: Zankov, Dmitry, Polishchuk, Pavlo, Sobieraj, Michal, Barbatti, Mario
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2512.01287
Tags: Add Tag
No Tags, Be the first to tag this record!
Table of Contents:
  • We introduce milearn, a Python package for multi-instance learning (MIL) that follows the familiar scikit-learn fit/predict interface while providing a unified framework for both classical and neural-network-based MIL algorithms for regression and classification. The package also includes built-in hyperparameter optimization designed specifically for small MIL datasets, enabling robust model selection in data-scarce scenarios. We demonstrate the versatility of milearn across a broad range of synthetic MIL benchmark datasets, including digit classification and regression, molecular property prediction, and protein-protein interaction (PPI) prediction. Special emphasis is placed on the key instance detection (KID) problem, for which the package provides dedicated support.