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 사이트를 제공합니다.

일반적인 작업 흐름

  1. 평소처럼 모델을 훈련합니다(예: LinearRegression, RandomForest, PyTorch 네트워크).
  2. MAPIE가 불확실성을 추정하는 데 사용할 홀드아웃 "교정" 세트를 분할합니다.
  3. 훈련된 모델을 MAPIE 클래스(MapieRegressor, MapieClassifier 등)로 래핑합니다.
  4. 교정 데이터로 래퍼를 피팅합니다.
  5. 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 패키지입니다. 특정 모델 아키텍처에 얽매이지 않고 신뢰할 수 있는 신뢰 구간이나 예측 집합이 필요할 때 유용합니다.

관련

  • 프로젝트
  • 프로젝트
  • 프로젝트
  • 프로젝트
  • 프로젝트