{"id":"https://openalex.org/W7167594551","doi":"https://doi.org/10.48550/arxiv.2607.05238","title":"Branch-JEPA: Finite-Support Predictive Distributions for JEPA World Models","display_name":"Branch-JEPA: Finite-Support Predictive Distributions for JEPA World Models","publication_year":2026,"publication_date":"2026-07-06","ids":{"openalex":"https://openalex.org/W7167594551","doi":"https://doi.org/10.48550/arxiv.2607.05238"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2607.05238","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2607.05238","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://doi.org/10.48550/arxiv.2607.05238","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5113237186","display_name":"Zhi Song","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Song, Zhi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5083473392","display_name":"Ximing Xing","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Xing, Ximing","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5140187295","display_name":"Zhenchao Tang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Tang, Zhenchao","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5122964295","display_name":"H Huang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Huang, hanbo","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5076907782","display_name":"Jiehui Huang","orcid":"https://orcid.org/0000-0002-3099-2886"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Huang, Jiehui","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5144101114","display_name":"Weilong Yan","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yan, Weilong","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5029935165","display_name":"Tianxu Lv","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Lv, Tianxu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5144313629","display_name":"Minghao Yang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yang, Minghao","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5135060419","display_name":"Zhongzheng Niu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Niu, Zhongzheng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5144386876","display_name":"Bing He","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"He, Bing","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5138407030","display_name":"lusheng wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Lusheng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5144345258","display_name":"Jianhua Yao","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yao, Jianhua","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":0,"corresponding_author_ids":[],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":null,"has_fulltext":false,"cited_by_count":0,"citation_normalized_percentile":null,"cited_by_percentile_year":null,"biblio":{"volume":null,"issue":null,"first_page":null,"last_page":null},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T13702","display_name":"Machine Learning in Healthcare","score":0.12770000100135803,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},"topics":[{"id":"https://openalex.org/T13702","display_name":"Machine Learning in Healthcare","score":0.12770000100135803,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.08460000157356262,"subfield":{"id":"https://openalex.org/subfields/1707","display_name":"Computer Vision and Pattern Recognition"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T12026","display_name":"Explainable Artificial Intelligence (XAI)","score":0.07400000095367432,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/successor-cardinal","display_name":"Successor cardinal","score":0.4668999910354614},{"id":"https://openalex.org/keywords/graph","display_name":"Graph","score":0.3776000142097473},{"id":"https://openalex.org/keywords/protocol","display_name":"Protocol (science)","score":0.34619998931884766},{"id":"https://openalex.org/keywords/markov-chain","display_name":"Markov chain","score":0.3156999945640564},{"id":"https://openalex.org/keywords/planner","display_name":"Planner","score":0.30559998750686646},{"id":"https://openalex.org/keywords/state","display_name":"State (computer science)","score":0.29190000891685486},{"id":"https://openalex.org/keywords/raw-data","display_name":"Raw data","score":0.28529998660087585}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.5349000096321106},{"id":"https://openalex.org/C75306776","wikidata":"https://www.wikidata.org/wiki/Q7632662","display_name":"Successor cardinal","level":2,"score":0.4668999910354614},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.39010000228881836},{"id":"https://openalex.org/C132525143","wikidata":"https://www.wikidata.org/wiki/Q141488","display_name":"Graph","level":2,"score":0.3776000142097473},{"id":"https://openalex.org/C2780385302","wikidata":"https://www.wikidata.org/wiki/Q367158","display_name":"Protocol (science)","level":3,"score":0.34619998931884766},{"id":"https://openalex.org/C98763669","wikidata":"https://www.wikidata.org/wiki/Q176645","display_name":"Markov chain","level":2,"score":0.3156999945640564},{"id":"https://openalex.org/C2776999362","wikidata":"https://www.wikidata.org/wiki/Q2349274","display_name":"Planner","level":2,"score":0.30559998750686646},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.30390000343322754},{"id":"https://openalex.org/C48103436","wikidata":"https://www.wikidata.org/wiki/Q599031","display_name":"State (computer science)","level":2,"score":0.29190000891685486},{"id":"https://openalex.org/C132964779","wikidata":"https://www.wikidata.org/wiki/Q2110223","display_name":"Raw data","level":2,"score":0.28529998660087585},{"id":"https://openalex.org/C2776359362","wikidata":"https://www.wikidata.org/wiki/Q2145286","display_name":"Representation (politics)","level":3,"score":0.2777000069618225},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.2750000059604645},{"id":"https://openalex.org/C45374587","wikidata":"https://www.wikidata.org/wiki/Q12525525","display_name":"Computation","level":2,"score":0.2712000012397766},{"id":"https://openalex.org/C114289077","wikidata":"https://www.wikidata.org/wiki/Q3284399","display_name":"Statistical model","level":2,"score":0.266400009393692},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.26600000262260437},{"id":"https://openalex.org/C81917197","wikidata":"https://www.wikidata.org/wiki/Q628760","display_name":"Selection (genetic algorithm)","level":2,"score":0.26460000872612},{"id":"https://openalex.org/C28719098","wikidata":"https://www.wikidata.org/wiki/Q44946","display_name":"Point (geometry)","level":2,"score":0.2644999921321869},{"id":"https://openalex.org/C198082294","wikidata":"https://www.wikidata.org/wiki/Q3399648","display_name":"Position (finance)","level":2,"score":0.2635999917984009},{"id":"https://openalex.org/C124101348","wikidata":"https://www.wikidata.org/wiki/Q172491","display_name":"Data mining","level":1,"score":0.2630999982357025},{"id":"https://openalex.org/C105795698","wikidata":"https://www.wikidata.org/wiki/Q12483","display_name":"Statistics","level":1,"score":0.2572999894618988},{"id":"https://openalex.org/C93959086","wikidata":"https://www.wikidata.org/wiki/Q6888345","display_name":"Model selection","level":2,"score":0.2549000084400177},{"id":"https://openalex.org/C159886148","wikidata":"https://www.wikidata.org/wiki/Q176645","display_name":"Markov process","level":2,"score":0.2549000084400177}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2607.05238","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2607.05238","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":null,"raw_source_name":null,"raw_type":"Preprint"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2607.05238","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2607.05238","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Joint-embedding":[0],"predictive":[1,202],"architectures":[2],"(JEPAs)":[3],"learn":[4],"dynamics":[5,30],"by":[6,100,106],"predicting":[7],"future":[8],"observations":[9],"in":[10,140],"representation":[11],"space.":[12],"Yet":[13],"most":[14],"JEPA":[15],"world":[16],"models":[17],"return":[18],"one":[19],"latent":[20,49,125,187],"successor,":[21],"even":[22],"when":[23],"hidden":[24],"intent,":[25],"partial":[26],"observation,":[27],"or":[28],"stochastic":[29],"make":[31],"several":[32],"futures":[33],"plausible.":[34],"We":[35],"introduce":[36],"Branch-JEPA,":[37],"which":[38],"replaces":[39],"this":[40],"point-valued":[41],"transition":[42],"with":[43,180],"a":[44,84,121],"context-weighted":[45],"finite":[46],"set":[47,59],"of":[48,199],"successors.":[50],"Every":[51],"branch":[52],"is":[53,60],"decoded":[54],"independently,":[55],"and":[56,76,102,111,132,138,183],"the":[57,89,149,173,197,200],"complete":[58],"retained":[61],"at":[62,148],"inference.":[63],"The":[64],"architecture":[65],"supports":[66],"two":[67],"complementary":[68],"training":[69,79,95],"regimes:":[70],"specialization":[71],"for":[72,80,172],"recovering":[73],"separated":[74],"successors":[75],"full-set":[77,94,194],"Energy-Score":[78],"distributional":[81],"fidelity.":[82],"In":[83,120,158],"locked":[85],"five-seed":[86],"evaluation":[87],"on":[88],"Argoverse~2":[90],"official":[91],"validation":[92],"split,":[93],"improves":[96,133,196],"trajectory":[97,104],"Energy":[98,134],"Score":[99],"$5.8$--$6.5\\%$":[101],"probability-weighted":[103],"distance":[105],"$9.3$--$10.4\\%$":[107],"over":[108,145],"matched-$K{=}6$":[109],"assignment":[110],"transport":[112],"objectives,":[113],"while":[114,193],"retaining":[115],"$5.36$":[116],"endpoint-deduplicated":[117],"effective":[118,130],"branches.":[119],"parameter-exact":[122],"official-validation":[123],"comparison,":[124],"branching":[126,146,188],"retains":[127],"$10.3\\%$":[128],"more":[129,190],"modes":[131],"Score,":[135],"expected":[136],"ADE,":[137],"Brier":[139],"all":[141],"five":[142],"paired":[143,153],"seeds":[144],"only":[147],"output":[150],"decoder;":[151],"every":[152],"95\\%":[154],"interval":[155],"excludes":[156],"zero.":[157],"an":[159],"OGBench":[160],"graph":[161],"audit,":[162],"Branch-JEPA":[163],"increases":[164],"teleport":[165],"verified-route":[166],"existence":[167],"to":[168],"$19.2\\%$":[169],"versus":[170],"$3.9\\%$":[171],"MDN.":[174],"Its":[175],"raw-support":[176],"advantage":[177],"also":[178],"persists":[179],"29-D":[181],"state":[182],"RGB":[184],"observations.":[185],"Together,":[186],"preserves":[189],"distinct":[191],"futures,":[192],"scoring":[195],"quality":[198],"resulting":[201],"distribution.":[203]},"counts_by_year":[],"updated_date":"2026-08-05T07:39:15.569665","created_date":"2026-07-08T00:00:00"}