본문으로 건너뛰기

트렌딩 - GitHub 인기 레포 & HuggingFace 모델

google-research/tabfm

Python1 / 0

TabFM은 사전학습된 가중치를 사용하고 학습 데이터를 컨텍스트로 읽어 즉석에서 혼합형 테이블의 분류와 회귀를 수행하는 scikit-learn 호환 라이브러리이다.

TL;DR

TabFM은 scikit-learn 호환 인터페이스를 제공하는 테이블 전용 파운데이션 모델로서 사전학습된 가중치를 로드하고 훈련 데이터를 컨텍스트로 읽어 별도 파인튜닝 없이 즉시 분류와 회귀를 수행한다. 사용자는 JAX 또는 PyTorch 백엔드 중 하나를 선택해 모델을 로드하고 TabFMClassifier나 TabFMRegressor를 통해 기존 scikit-learn 워크플로와 연동할 수 있다. 이 접근은 초기 프로토타이핑과 리소스가 제한된 환경에서 모델 학습 비용을 줄이는 데 유리하지만, 도메인 특화 성능 향상을 목적으로 하는 경우에는 추가적인 튜닝이 필요할 수 있다. README는 v1.0.0 가중치의 자동 다운로드, 혼합형 컬럼 전처리 준비, 예제 스크립트와 테스트 실행법을 명확히 안내하여 빠른 실험 진입을 지원한다.

핵심 포인트

  • scikit-learn 호환 인터페이스를 기본으로 제공하여 기존 파이프라인과의 통합 비용을 낮춘다. TabFMClassifier와 TabFMRegressor는 fit/predict/predict_proba 메서드를 제공하므로 기존 cross-validation 및 파이프라인 코드와 바로 연동이 가능하다. 이 점은 모델 교체나 프로토타이핑 단계에서 전환 비용을 크게 줄인다.
  • 훈련 데이터 자체를 컨텍스트로 읽어 제로샷 예측을 수행하는 운영 방식을 채택하여 데이터셋별 파인튜닝이不要한 워크플로를 만든다. 이 방식은 별도 GPU 자원을 들여 재학습하지 않고도 다양한 테이블 형태에 실험적으로 적용할 수 있게 한다. 다만 데이터셋 특화 성능 향상이 필요한 경우 별도 파인튜닝을 적용할 수 없다면 한계가 존재한다.
  • JAX와 PyTorch 두 가지 백엔드를 공식적으로 지원하여 다양한 실행 환경에 적응할 수 있다. README는 JAX(CPU/GPU)와 PyTorch 설치 옵션 및 핵심 버전을 명시하여 사용자가 환경에 맞는 백엔드를 선택할 수 있도록 설계되었다. 이로 인해 특정 하드웨어나 프레임워크 의존성에 맞춘 배포가 용이하다.
  • 사전학습 가중치를 자동으로 다운로드하고 JAX 또는 PyTorch 백엔드로 로드하여 별도 파인튜닝 없이 즉시 추론을 실행할 수 있다. 이 동작은 TabFM v1.0.0 릴리스에서 제공되는 모델 로드 함수를 통해 수행되며, 사용자는 백엔드만 선택하면 된다. scikit-learn 호환 래퍼는 기존 파이프라인과의 통합을 용이하게 한다.

2.5k

STARS

255

FORKS

+205

TRENDING

1

조회수

watchers 2.5kopen issues 40Apache License 2.0

관련 토론

아직 관련 토론이 없습니다.

댓글

댓글을 작성하려면 로그인이 필요합니다.