跳到正文
IFAI

開源專案

JEPA-Anything:跨領域通用預測框架,用正交分解拆解狀態推演

首次發布 傳播指數 48

開源框架 JEPA-Anything 試圖用統一邏輯搞定不同學科的預測推演。它基於 JEPA 架構提出「正交預測分解」,將複雜狀態拆解為獨立要素分頭推算再拼合,各領域僅需適配專屬介面即可複用核心預測機制。目前倉庫提供程式碼腳手架與合成示例,未包含預訓練權重。

JEPA-Anything:跨領域通用預測框架,用正交分解拆解狀態推演
圖:github.com

JEPA-Anything 是一個跨領域的通用預測框架,能幫你為生物、物理、醫療等不同領域構建統一的世界預測模型。

不管是單細胞變化、分子動力學演化,還是臨床醫療軌跡分析,本質上都在做一件事:根據當前已知狀態推測未來的演變。JEPA-Anything 試圖用同一套學習邏輯,搞定不同學科裡的預測推演。

它是怎麼做到的

傳統基於自監督或 JEPA 架構的模型,往往把要預測的目標當成一個不可分割的整體去硬猜。這種做法遇到複雜的多變數系統,很容易顧此失彼。

JEPA-Anything 的核心機制叫「正交預測分解」(Orthogonal Predictive Factorization,簡稱 OPF)。模型會先用編碼器把觀測資料對映到潛在世界狀態,再把這個狀態拆解成若干組互不干擾的預測座標要素。

每個要素都走獨立的預測通道推演變化,最後再重新拼回完整的潛在狀態。拼合後的狀態可以直接用於下游的屬性讀出、干預模擬、決策規劃或多步長期推演。

各個領域的資料形態天差地別,所以每個學科依舊保留專屬的資料觀察介面和評估介面卡,但核心的預測運算元與狀態介面完全共享。換句話說,不用為每個領域從頭重寫一套預測機制。

怎麼跑起來與當前限制

這個專案目前在 GitHub 開源,環境要求 Python 3.10 或更高版本。核心預測庫基於 PyTorch,而任務設計工具和示例只依賴 Python 標準庫。

克隆倉庫後,通過 pip 安裝核心庫,就能執行內建的合成線性動力學示例。工具包裡還包含了設計校驗工具和腳手架生成指令碼,可以用 JSON 格式配置任務規範,一鍵生成介面卡、測試和模型骨架。

必須說清楚的是,倉庫目前提供的是通用的核心程式碼、腳手架工具以及用來跑通流程的合成數據示例。官方並沒有放出在單細胞或臨床資料集上訓練好的大模型權重,圖表中展示的生物學對應關係需要開發者自己用實驗資料去訓練和驗證。

想要試用的話,可以訪問 GitHub 上的 Gen-Verse/JEPA-Anything 倉庫,按照文件在 Linux 或 macOS 終端中完成腳手架部署。

為什麼重要

我們認為 JEPA-Anything 最聰明的地方,在於它沒有嘗試做一個包打天下的巨型端到端模型,而是把各領域的介面抽象出來,只統一底層的預測流。

對於從事 AI for Science、生物計算或複雜物理系統建模的研究者來說,它提供了一套現成的架構規範,省去了自己從零搭建潛在預測通道的重複勞動。不過在缺乏官方預訓練權重的情況下,門檻仍然不低,普通開發者很難直接開箱即用,它更適合有自有領域資料並打算落地世界模型的科研團隊。

來源 (1)

  1. [1]

每日 AI 內容精選