English
Related papers

Related papers: torchmil: A PyTorch-based library for deep Multipl…

200 papers

Multiple Instance Learning (MIL) has enabled weakly supervised analysis of whole-slide images (WSIs) in computational pathology. However, traditional MIL approaches often lose crucial contextual information, while transformer-based…

Computer Vision and Pattern Recognition · Computer Science 2025-12-30 Xitong Ling , Minxi Ouyang , Xiaoxiao Li , Jiawen Li , Ying Chen , Yuxuan Sun , Xinrui Chen , Tian Guan , Xiaoping Liu , Yonghong He

DoubleML is an open-source Python library implementing the double machine learning framework of Chernozhukov et al. (2018) for a variety of causal models. It contains functionalities for valid statistical inference on causal parameters when…

Machine Learning · Statistics 2022-10-06 Philipp Bach , Victor Chernozhukov , Malte S. Kurz , Martin Spindler

In this paper, we introduce McTorch, a manifold optimization library for deep learning that extends PyTorch. It aims to lower the barrier for users wishing to use manifold constraints in deep learning applications, i.e., when the parameters…

Machine Learning · Statistics 2018-10-05 Mayank Meghwanshi , Pratik Jawanpuria , Anoop Kunchukuttan , Hiroyuki Kasai , Bamdev Mishra

Continual Learning is an important and challenging problem in machine learning, where models must adapt to a continuous stream of new data without forgetting previously acquired knowledge. While existing frameworks are built on PyTorch, the…

Machine Learning · Computer Science 2023-04-24 Nikolaos Dimitriadis , Francois Fleuret , Pascal Frossard

We design and implement a ready-to-use library in PyTorch for performing micro-batch pipeline parallelism with checkpointing proposed by GPipe (Huang et al., 2019). In particular, we develop a set of design components to enable…

Distributed, Parallel, and Cluster Computing · Computer Science 2020-04-22 Chiheon Kim , Heungsub Lee , Myungryong Jeong , Woonhyuk Baek , Boogeon Yoon , Ildoo Kim , Sungbin Lim , Sungwoong Kim

YAMLE: Yet Another Machine Learning Environment is an open-source framework that facilitates rapid prototyping and experimentation with machine learning (ML) models and methods. The key motivation is to reduce repetitive work when…

Machine Learning · Computer Science 2024-02-12 Martin Ferianc , Miguel Rodrigues

MIML library is a Java software tool to develop, test, and compare classification algorithms for multi-instance multi-label (MIML) learning. The library includes 43 algorithms and provides a specific format and facilities for data managing…

Machine Learning · Computer Science 2024-02-14 Álvaro Belmonte , Amelia Zafra , Eva Gibaja

The field of Multi-Agent Reinforcement Learning (MARL) is currently facing a reproducibility crisis. While solutions for standardized reporting have been proposed to address the issue, we still lack a benchmarking tool that enables…

Machine Learning · Computer Science 2024-11-26 Matteo Bettini , Amanda Prorok , Vincent Moens

modAL is a modular active learning framework for Python, aimed to make active learning research and practice simpler. Its distinguishing features are (i) clear and modular object oriented design (ii) full compatibility with scikit-learn…

Machine Learning · Computer Science 2018-12-13 Tivadar Danka , Peter Horvath

We present \texttt{secml}, an open-source Python library for secure and explainable machine learning. It implements the most popular attacks against machine learning, including test-time evasion attacks to generate adversarial examples…

Machine Learning · Computer Science 2022-05-16 Maura Pintor , Luca Demetrio , Angelo Sotgiu , Marco Melis , Ambra Demontis , Battista Biggio

Interventions on model-internal states are fundamental operations in many areas of AI, including model editing, steering, robustness, and interpretability. To facilitate such research, we introduce $\textbf{pyvene}$, an open-source Python…

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…

Machine Learning · Computer Science 2025-12-02 Dmitry Zankov , Pavlo Polishchuk , Michal Sobieraj , Mario Barbatti

Computational pathology holds substantial promise for improving diagnosis and guiding treatment decisions. Recent pathology foundation models enable the extraction of rich patch-level representations from large-scale whole-slide images…

Computer Vision and Pattern Recognition · Computer Science 2025-11-20 Xiangde Luo , Jinxi Xiang , Yuanfeng Ji , Ruijiang Li

Multiple instance learning (MIL) is a variation of traditional supervised learning problems where data (referred to as bags) are composed of sub-elements (referred to as instances) and only bag labels are available. MIL has a variety of…

Machine Learning · Computer Science 2018-05-02 Han Bao , Tomoya Sakai , Issei Sato , Masashi Sugiyama

Mixed Integer Linear Programming (MILP) is essential for modeling complex decision-making problems but faces challenges in computational tractability and requires expert formulation. Current deep learning approaches for MILP focus on…

Machine Learning · Computer Science 2025-02-24 Sirui Li , Janardhan Kulkarni , Ishai Menache , Cathy Wu , Beibin Li

Finding errors in machine learning applications requires a thorough exploration of their behavior over data. Existing approaches used by practitioners are often ad-hoc and lack the abstractions needed to scale this process. We present…

Databases · Computer Science 2024-10-17 Aaditya Naik , Adam Stein , Yinjun Wu , Mayur Naik , Eric Wong

A new random forest based model for solving the Multiple Instance Learning (MIL) problem under small tabular data, called Soft Tree Ensemble MIL (STE-MIL), is proposed. A new type of soft decision trees is considered, which is similar to…

Machine Learning · Computer Science 2023-02-14 Andrei V. Konstantinov , Lev V. Utkin

We introduce rLLM (relationLLM), a PyTorch library designed for Relational Table Learning (RTL) with Large Language Models (LLMs). The core idea is to decompose state-of-the-art Graph Neural Networks, LLMs, and Table Neural Networks into…

Artificial Intelligence · Computer Science 2025-11-13 Weichen Li , Xiaotong Huang , Jianwu Zheng , Zheng Wang , Chaokun Wang , Li Pan , Jianhua Li

Various multi-instance learning (MIL) based approaches have been developed and successfully applied to whole-slide pathological images (WSI). Existing MIL methods emphasize the importance of feature aggregators, but largely neglect the…

Computer Vision and Pattern Recognition · Computer Science 2025-05-13 Yicheng Song , Tiancheng Lin , Die Peng , Su Yang , Yi Xu

Multiple instance learning (MIL) is a robust paradigm for whole-slide pathological image (WSI) analysis, processing gigapixel-resolution images with slide-level labels. As pioneering efforts, attention-based MIL (ABMIL) and its variants are…

Computer Vision and Pattern Recognition · Computer Science 2024-04-02 Linghan Cai , Shenjin Huang , Ye Zhang , Jinpeng Lu , Yongbing Zhang