scikit-learn-contrib/MAPIE
A scikit-learn-compatible library for estimating prediction intervals and controlling risks, based on conformal predictions.
MAPIE – 모델 비의존적 예측 구간 추정기
정의
- 기존 모델 위에 예측 구간(회귀용) 및 예측 집합(분류, 시계열, 세그멘테이션 등)을 추가하는 Python 라이브러리입니다.
- scikit-learn 추정기와 즉시 호환되며, TensorFlow 또는 PyTorch 모델을 위한 얇은 래퍼도 제공하므로 기존 도구를 그대로 사용할 수 있습니다.
중요성
- 많은 실제 애플리케이션에서는 점(point) 예측뿐만 아니라 그 예측의 불확실성에 대한 신뢰할 수 있는 정보가 필요합니다. MAPIE는 컨포멀 예측 및 관련 위험 제어 이론에 기반한 통계적으로 타당하고 분포에 의존하지 않는 불확실성 추정을 제공합니다.
- 이러한 보장은 매우 완화된 가정(본질적으로 교정(calibration) 데이터가 교환 가능(exchangeable)하다는 가정) 하에서 성립하므로, 표 형식 회귀부터 이미지 세그멘테이션, 그리고 새로운 LLM-as-judge 시나리오에 이르기까지 다양한 도메인에서 유용합니다.
주요 기능 (README에 설명됨)
- 예측 구간/집합: 회귀, 분류, 시계열, 다중 레이블 분류, 의미론적 세그멘테이션.
- 위험 제어: 목표 위험(예: 최대 거짓 음성 비율)을 설정하고 해당 위험을 증명 가능한 수준으로 충족하는 예측 집합을 얻을 수 있습니다.
- 모델 비의존: scikit-learn 호환 추정기와 함께 작동하며, 선택적 래퍼를 통해 TensorFlow 또는 PyTorch 모델을 연결할 수 있습니다.
- 적응형 및 교환성 테스트: 2026년의 새로운 기능으로 교환성 가정이 성립하는지 테스트하는 방법과 가능할 때 구간을 좁히는 적응형 컨포멀 기법이 포함됩니다.
- 광범위한 문서 및 예제: 회귀 및 분류를 위한 빠른 시작 노트북과 완전한 Read-the-Docs 사이트를 제공합니다.
일반적인 작업 흐름
- 평소처럼 모델을 훈련합니다(예:
LinearRegression,RandomForest, PyTorch 네트워크). - MAPIE가 불확실성을 추정하는 데 사용할 홀드아웃 "교정" 세트를 분할합니다.
- 훈련된 모델을 MAPIE 클래스(
MapieRegressor,MapieClassifier등)로 래핑합니다. - 교정 데이터로 래퍼를 피팅합니다.
predict_interval또는predict_set로 예측하여 일반적인 점 예측과 함께 구간/집합을 얻습니다.
설치
pip install mapie # 또는 conda install -c conda-forge mapie
요구 사항: Python ≥ 3.9, NumPy ≥ 1.23, scikit-learn ≥ 1.4.
커뮤니티 및 지속 가능성
- scikit-learn-contrib 생태계의 일부로, 동일한 오픈소스 거버넌스 및 테스트 표준을 따릅니다.
- 활발히 유지 관리됩니다(GitHub Actions CI, 코드 커버리지 배지, 정기 릴리스).
- 학계 및 산업 파트너(Capgemini Invent, INRA, Michelin, ENS Paris-Saclay 등)의 지원을 받으며, 지역 및 민간 연구 기관의 자금을 지원받습니다.
MAPIE를 사용해야 하는 경우
- 이미 보유한 모델에 대해 교정된 불확실성이 필요한 경우.
- 애플리케이션이 공식적인 위험 보장을 요구하는 경우(예: 의료 진단, 자율 주행, 금융 예측).
- 특정 모델 패밀리를 가정하지 않고 최소한의 추가 코드로 작동하는 라이브러리를 선호하는 경우.
자세히 알아보기
- 문서의 빠른 시작 노트북: 회귀 예제, 분류 예제.
- Read-the-Docs 사이트에서 전체 API 참조 및 이론적 배경을 확인하세요.
- README에 인용된 학술 논문을 통해 더 깊은 통계적 통찰력을 얻을 수 있습니다.
TL;DR: MAPIE는 컨포멀 예측 및 위험 제어 방법을 사용하여 모든 ML 모델에 통계적으로 엄격한 불확실성 정량화를 추가하는, 잘 유지 관리되는 모델 비의존 Python 패키지입니다. 특정 모델 아키텍처에 얽매이지 않고 신뢰할 수 있는 신뢰 구간이나 예측 집합이 필요할 때 유용합니다.
관련
- 프로젝트
- 프로젝트
- 프로젝트
- 프로젝트
- 프로젝트