금융 캔들스틱 전용 오픈소스 AI 모델 Kronos, AAAI 2026 채택되며 깃허브 스타 11,800개 돌파

금융 캔들스틱(K-line) 데이터에 특화된 최초의 오픈소스 파운데이션 모델 Kronos가 깃허브 트렌딩에 진입했다. 현재 스타 11,890개를 기록하고 있으며, 하루에만 223개의 스타를 추가로 획득했다.

Kronos는 45개 이상의 글로벌 거래소 데이터로 사전 학습된 디코더 전용(decoder-only) 파운데이션 모델 패밀리이다. 범용 시계열 모델과 달리, 금융 데이터 특유의 높은 노이즈 특성을 처리하도록 설계되었다.

핵심 아키텍처는 2단계 프레임워크로 구성된다. 1단계에서 전용 토크나이저가 연속적이고 다차원적인 K-line 데이터(OHLCV: 시가, 고가, 저가, 종가, 거래량)를 계층적 이산 토큰으로 양자화한다. 2단계에서 대규모 오토리그레시브 트랜스포머가 이 토큰들로 사전 학습되어 다양한 퀀트 업무에 활용할 수 있는 통합 모델로 작동한다.

모델은 네 가지 크기로 제공된다. Kronos-mini(파라미터 4.1M, 컨텍스트 길이 2,048), Kronos-small(24.7M, 512), Kronos-base(102.3M, 512)는 Hugging Face Hub에서 오픈소스로 공개되어 있고, Kronos-large(499.2M)는 비공개 상태이다.

이 모델은 AAAI 2026 학회에 채택되었으며, 논문은 arXiv에서 확인할 수 있다. BTC/USDT 거래 쌍의 향후 24시간 예측을 보여주는 라이브 데모도 제공하고 있다.

사용 방법은 간단하다. KronosPredictor 클래스를 통해 데이터 전처리, 정규화, 예측, 역정규화까지 몇 줄의 코드로 처리할 수 있다. 단일 시계열 예측뿐 아니라, 여러 자산이나 기간을 동시에 예측하는 배치 예측 기능도 지원하며 GPU 병렬 처리를 활용한다.

파인튜닝 파이프라인도 함께 공개되어 있다. Qlib를 활용한 중국 A주 시장 데이터 준비, 토크나이저와 예측기 모델의 파인튜닝, 백테스팅으로 이어지는 4단계 과정을 예시로 제공한다.

개발팀은 해당 파인튜닝 파이프라인이 시연 목적의 단순화된 예시이며, 실제 프로덕션 수준의 퀀트 트레이딩 시스템에는 포트폴리오 최적화와 리스크 팩터 중립화 등 보다 정교한 기법이 필요하다고 밝혔다.