{"id":"https://openalex.org/W4281953044","doi":"https://doi.org/10.48550/arxiv.2205.13603","title":"Tensor Program Optimization with Probabilistic Programs","display_name":"Tensor Program Optimization with Probabilistic Programs","publication_year":2022,"publication_date":"2022-05-26","ids":{"openalex":"https://openalex.org/W4281953044","doi":"https://doi.org/10.48550/arxiv.2205.13603"},"language":"en","primary_location":{"id":"pmh:oai:arXiv.org:2205.13603","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2205.13603","pdf_url":"https://arxiv.org/pdf/2205.13603","source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"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":null,"license_id":null,"version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},"type":"preprint","indexed_in":["arxiv","datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://arxiv.org/pdf/2205.13603","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5022832599","display_name":"Junru Shao","orcid":"https://orcid.org/0000-0002-7370-1495"},"institutions":[],"countries":[],"is_corresponding":true,"raw_author_name":"Shao, Junru","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5069002442","display_name":"Xiyou Zhou","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhou, Xiyou","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5053373302","display_name":"Siyuan Feng","orcid":"https://orcid.org/0000-0003-2531-8480"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Feng, Siyuan","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5047212151","display_name":"Bohan Hou","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Hou, Bohan","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5025482495","display_name":"Ruihang Lai","orcid":"https://orcid.org/0000-0001-6400-5079"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Lai, Ruihang","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5101264780","display_name":"Hongyi Jin","orcid":"https://orcid.org/0000-0001-6894-6554"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Jin, Hongyi","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5076540989","display_name":"Wuwei Lin","orcid":"https://orcid.org/0000-0001-8589-5453"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Lin, Wuwei","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5046294658","display_name":"Masahiro Masuda","orcid":"https://orcid.org/0009-0006-1054-3836"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Masuda, Masahiro","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5008633310","display_name":"Cody Hao Yu","orcid":"https://orcid.org/0000-0002-9298-6254"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yu, Cody Hao","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5101471083","display_name":"Tianqi Chen","orcid":"https://orcid.org/0000-0002-2336-1875"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Chen, Tianqi","raw_affiliation_strings":[],"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":10,"corresponding_author_ids":["https://openalex.org/A5022832599"],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":null,"has_fulltext":false,"cited_by_count":10,"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/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.9955000281333923,"subfield":{"id":"https://openalex.org/subfields/1708","display_name":"Hardware and Architecture"},"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/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.9955000281333923,"subfield":{"id":"https://openalex.org/subfields/1708","display_name":"Hardware and Architecture"},"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/T13650","display_name":"Computational Physics and Python Applications","score":0.989300012588501,"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/T12303","display_name":"Tensor decomposition and applications","score":0.9771999716758728,"subfield":{"id":"https://openalex.org/subfields/2605","display_name":"Computational Mathematics"},"field":{"id":"https://openalex.org/fields/26","display_name":"Mathematics"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/computer-science","display_name":"Computer science","score":0.7469199895858765},{"id":"https://openalex.org/keywords/probabilistic-logic","display_name":"Probabilistic logic","score":0.6508424878120422},{"id":"https://openalex.org/keywords/domain","display_name":"Domain (mathematical analysis)","score":0.6396709680557251},{"id":"https://openalex.org/keywords/abstraction","display_name":"Abstraction","score":0.6311770081520081},{"id":"https://openalex.org/keywords/space","display_name":"Space (punctuation)","score":0.5951812267303467},{"id":"https://openalex.org/keywords/modular-design","display_name":"Modular design","score":0.5565377473831177},{"id":"https://openalex.org/keywords/theoretical-computer-science","display_name":"Theoretical computer science","score":0.5435798168182373},{"id":"https://openalex.org/keywords/tensor","display_name":"Tensor (intrinsic definition)","score":0.5111833214759827},{"id":"https://openalex.org/keywords/speedup","display_name":"Speedup","score":0.45951855182647705},{"id":"https://openalex.org/keywords/transformation","display_name":"Transformation (genetics)","score":0.43955400586128235},{"id":"https://openalex.org/keywords/state-space","display_name":"State space","score":0.43303200602531433},{"id":"https://openalex.org/keywords/artificial-intelligence","display_name":"Artificial intelligence","score":0.3583482503890991},{"id":"https://openalex.org/keywords/programming-language","display_name":"Programming language","score":0.26641395688056946},{"id":"https://openalex.org/keywords/mathematics","display_name":"Mathematics","score":0.13154983520507812},{"id":"https://openalex.org/keywords/parallel-computing","display_name":"Parallel computing","score":0.07095843553543091}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7469199895858765},{"id":"https://openalex.org/C49937458","wikidata":"https://www.wikidata.org/wiki/Q2599292","display_name":"Probabilistic logic","level":2,"score":0.6508424878120422},{"id":"https://openalex.org/C36503486","wikidata":"https://www.wikidata.org/wiki/Q11235244","display_name":"Domain (mathematical analysis)","level":2,"score":0.6396709680557251},{"id":"https://openalex.org/C124304363","wikidata":"https://www.wikidata.org/wiki/Q673661","display_name":"Abstraction","level":2,"score":0.6311770081520081},{"id":"https://openalex.org/C2778572836","wikidata":"https://www.wikidata.org/wiki/Q380933","display_name":"Space (punctuation)","level":2,"score":0.5951812267303467},{"id":"https://openalex.org/C101468663","wikidata":"https://www.wikidata.org/wiki/Q1620158","display_name":"Modular design","level":2,"score":0.5565377473831177},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.5435798168182373},{"id":"https://openalex.org/C155281189","wikidata":"https://www.wikidata.org/wiki/Q3518150","display_name":"Tensor (intrinsic definition)","level":2,"score":0.5111833214759827},{"id":"https://openalex.org/C68339613","wikidata":"https://www.wikidata.org/wiki/Q1549489","display_name":"Speedup","level":2,"score":0.45951855182647705},{"id":"https://openalex.org/C204241405","wikidata":"https://www.wikidata.org/wiki/Q461499","display_name":"Transformation (genetics)","level":3,"score":0.43955400586128235},{"id":"https://openalex.org/C72434380","wikidata":"https://www.wikidata.org/wiki/Q230930","display_name":"State space","level":2,"score":0.43303200602531433},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.3583482503890991},{"id":"https://openalex.org/C199360897","wikidata":"https://www.wikidata.org/wiki/Q9143","display_name":"Programming language","level":1,"score":0.26641395688056946},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.13154983520507812},{"id":"https://openalex.org/C173608175","wikidata":"https://www.wikidata.org/wiki/Q232661","display_name":"Parallel computing","level":1,"score":0.07095843553543091},{"id":"https://openalex.org/C134306372","wikidata":"https://www.wikidata.org/wiki/Q7754","display_name":"Mathematical analysis","level":1,"score":0.0},{"id":"https://openalex.org/C111919701","wikidata":"https://www.wikidata.org/wiki/Q9135","display_name":"Operating system","level":1,"score":0.0},{"id":"https://openalex.org/C105795698","wikidata":"https://www.wikidata.org/wiki/Q12483","display_name":"Statistics","level":1,"score":0.0},{"id":"https://openalex.org/C185592680","wikidata":"https://www.wikidata.org/wiki/Q2329","display_name":"Chemistry","level":0,"score":0.0},{"id":"https://openalex.org/C55493867","wikidata":"https://www.wikidata.org/wiki/Q7094","display_name":"Biochemistry","level":1,"score":0.0},{"id":"https://openalex.org/C104317684","wikidata":"https://www.wikidata.org/wiki/Q7187","display_name":"Gene","level":2,"score":0.0},{"id":"https://openalex.org/C202444582","wikidata":"https://www.wikidata.org/wiki/Q837863","display_name":"Pure mathematics","level":1,"score":0.0},{"id":"https://openalex.org/C111472728","wikidata":"https://www.wikidata.org/wiki/Q9471","display_name":"Epistemology","level":1,"score":0.0},{"id":"https://openalex.org/C138885662","wikidata":"https://www.wikidata.org/wiki/Q5891","display_name":"Philosophy","level":0,"score":0.0}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:oai:arXiv.org:2205.13603","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2205.13603","pdf_url":"https://arxiv.org/pdf/2205.13603","source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"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":null,"license_id":null,"version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},{"id":"doi:10.48550/arxiv.2205.13603","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2205.13603","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"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":null,"license_id":null,"version":null,"is_accepted":false,"is_published":null,"raw_source_name":null,"raw_type":"article"}],"best_oa_location":{"id":"pmh:oai:arXiv.org:2205.13603","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2205.13603","pdf_url":"https://arxiv.org/pdf/2205.13603","source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"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":null,"license_id":null,"version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":["https://openalex.org/W2058965144","https://openalex.org/W2164382479","https://openalex.org/W2146343568","https://openalex.org/W98480971","https://openalex.org/W2150291671","https://openalex.org/W2013643406","https://openalex.org/W2027972911","https://openalex.org/W2157978810","https://openalex.org/W1822289477","https://openalex.org/W273478400"],"abstract_inverted_index":{"Automatic":[0],"optimization":[1,18,124],"for":[2,103],"tensor":[3,66,122],"programs":[4],"becomes":[5],"increasingly":[6],"important":[7],"as":[8],"we":[9],"deploy":[10],"deep":[11,152],"learning":[12,153],"in":[13,82,119,126],"various":[14],"environments,":[15],"and":[16,25,77,141],"efficient":[17],"relies":[19],"on":[20,150],"a":[21,32,53,61,83,104,127],"rich":[22,62],"search":[23,33,47,63,106,116,139],"space":[24,34,64,117,140],"effective":[26],"search.":[27],"Most":[28],"existing":[29],"efforts":[30],"adopt":[31],"which":[35,146],"lacks":[36],"the":[37,46,75,115,120,138,144],"ability":[38],"to":[39,44,59,73,86,98,135],"efficiently":[40],"enable":[41],"domain":[42,71,133],"experts":[43,72,134],"grow":[45,137],"space.":[48,107],"This":[49],"paper":[50],"introduces":[51],"MetaSchedule,":[52],"domain-specific":[54],"probabilistic":[55],"programming":[56],"language":[57],"abstraction":[58,69],"construct":[60],"of":[65],"programs.":[67],"Our":[68],"allows":[70],"analyze":[74],"program,":[76],"easily":[78],"propose":[79],"stochastic":[80],"choices":[81],"modular":[84,128],"way":[85],"compose":[87],"program":[88,102,123],"transformation":[89],"accordingly.":[90],"We":[91],"also":[92],"build":[93],"an":[94,100],"end-to-end":[95,151],"learning-driven":[96],"framework":[97],"find":[99],"optimized":[101],"given":[105],"Experimental":[108],"results":[109],"show":[110],"that":[111],"MetaSchedule":[112],"can":[113],"cover":[114],"used":[118],"state-of-the-art":[121],"frameworks":[125],"way.":[129],"Additionally,":[130],"it":[131],"empowers":[132],"conveniently":[136],"modularly":[142],"enhance":[143],"system,":[145],"brings":[147],"48%":[148],"speedup":[149],"workloads.":[154]},"counts_by_year":[{"year":2025,"cited_by_count":6},{"year":2024,"cited_by_count":3},{"year":2023,"cited_by_count":1}],"updated_date":"2026-03-20T23:20:44.827607","created_date":"2025-10-10T00:00:00"}