{"id":"https://openalex.org/W4200077405","doi":"https://doi.org/10.1109/ictai52525.2021.00045","title":"Bansor: Improving Tensor Program Auto-Scheduling with Bandit Based Reinforcement Learning","display_name":"Bansor: Improving Tensor Program Auto-Scheduling with Bandit Based Reinforcement Learning","publication_year":2021,"publication_date":"2021-11-01","ids":{"openalex":"https://openalex.org/W4200077405","doi":"https://doi.org/10.1109/ictai52525.2021.00045"},"language":"en","primary_location":{"id":"doi:10.1109/ictai52525.2021.00045","is_oa":false,"landing_page_url":"https://doi.org/10.1109/ictai52525.2021.00045","pdf_url":null,"source":{"id":"https://openalex.org/S4363608052","display_name":"2021 IEEE 33rd International Conference on Tools with Artificial Intelligence (ICTAI)","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"conference"},"license":null,"license_id":null,"version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"2021 IEEE 33rd International Conference on Tools with Artificial Intelligence (ICTAI)","raw_type":"proceedings-article"},"type":"conference-paper","indexed_in":["crossref"],"open_access":{"is_oa":false,"oa_status":"closed","oa_url":null,"any_repository_has_fulltext":false},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5086236078","display_name":"Chao Gao","orcid":"https://orcid.org/0000-0002-5865-2285"},"institutions":[{"id":"https://openalex.org/I4210115038","display_name":"Huawei Technologies (Canada)","ror":"https://ror.org/026venb53","country_code":"CA","type":"company","lineage":["https://openalex.org/I2250955327","https://openalex.org/I4210115038"]}],"countries":["CA"],"is_corresponding":false,"raw_author_name":"Chao Gao","raw_affiliation_strings":["Huawei Research Institute, Canada"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Huawei Research Institute, Canada","institution_ids":["https://openalex.org/I4210115038"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5059356240","display_name":"Tong Mo","orcid":"https://orcid.org/0000-0002-3564-4610"},"institutions":[{"id":"https://openalex.org/I4210115038","display_name":"Huawei Technologies (Canada)","ror":"https://ror.org/026venb53","country_code":"CA","type":"company","lineage":["https://openalex.org/I2250955327","https://openalex.org/I4210115038"]}],"countries":["CA"],"is_corresponding":false,"raw_author_name":"Tong Mo","raw_affiliation_strings":["Huawei Research Institute, Canada"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Huawei Research Institute, Canada","institution_ids":["https://openalex.org/I4210115038"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5015436290","display_name":"Taylor Zowtuk","orcid":null},"institutions":[{"id":"https://openalex.org/I4210115038","display_name":"Huawei Technologies (Canada)","ror":"https://ror.org/026venb53","country_code":"CA","type":"company","lineage":["https://openalex.org/I2250955327","https://openalex.org/I4210115038"]}],"countries":["CA"],"is_corresponding":false,"raw_author_name":"Taylor Zowtuk","raw_affiliation_strings":["Huawei Research Institute, Canada"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Huawei Research Institute, Canada","institution_ids":["https://openalex.org/I4210115038"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5054688856","display_name":"Tanvir Sajed","orcid":"https://orcid.org/0009-0002-6692-4161"},"institutions":[{"id":"https://openalex.org/I4210115038","display_name":"Huawei Technologies (Canada)","ror":"https://ror.org/026venb53","country_code":"CA","type":"company","lineage":["https://openalex.org/I2250955327","https://openalex.org/I4210115038"]}],"countries":["CA"],"is_corresponding":false,"raw_author_name":"Tanvir Sajed","raw_affiliation_strings":["Huawei Research Institute, Canada"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Huawei Research Institute, Canada","institution_ids":["https://openalex.org/I4210115038"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5053438368","display_name":"Laiyuan Gong","orcid":null},"institutions":[{"id":"https://openalex.org/I2250955327","display_name":"Huawei Technologies (China)","ror":"https://ror.org/00cmhce21","country_code":"CN","type":"company","lineage":["https://openalex.org/I2250955327"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Laiyuan Gong","raw_affiliation_strings":["Huawei Technologies, China"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Huawei Technologies, China","institution_ids":["https://openalex.org/I2250955327"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5020024289","display_name":"Hanxuan Chen","orcid":null},"institutions":[{"id":"https://openalex.org/I2250955327","display_name":"Huawei Technologies (China)","ror":"https://ror.org/00cmhce21","country_code":"CN","type":"company","lineage":["https://openalex.org/I2250955327"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Hanxuan Chen","raw_affiliation_strings":["Huawei Technologies, China"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Huawei Technologies, China","institution_ids":["https://openalex.org/I2250955327"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5012650575","display_name":"Shangling Jui","orcid":"https://orcid.org/0000-0002-1047-4264"},"institutions":[{"id":"https://openalex.org/I2250955327","display_name":"Huawei Technologies (China)","ror":"https://ror.org/00cmhce21","country_code":"CN","type":"company","lineage":["https://openalex.org/I2250955327"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Shangling Jui","raw_affiliation_strings":["Huawei Technologies, China"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Huawei Technologies, China","institution_ids":["https://openalex.org/I2250955327"]}]},{"author_position":"last","author":{"id":"https://openalex.org/A5100766134","display_name":"Wei Lu","orcid":"https://orcid.org/0000-0002-5046-4001"},"institutions":[{"id":"https://openalex.org/I4210115038","display_name":"Huawei Technologies (Canada)","ror":"https://ror.org/026venb53","country_code":"CA","type":"company","lineage":["https://openalex.org/I2250955327","https://openalex.org/I4210115038"]}],"countries":["CA"],"is_corresponding":false,"raw_author_name":"Wei Lu","raw_affiliation_strings":["Huawei Research Institute, Canada"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Huawei Research Institute, Canada","institution_ids":["https://openalex.org/I4210115038"]}]}],"institutions":[],"countries_distinct_count":2,"institutions_distinct_count":2,"corresponding_author_ids":[],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":0.6761,"has_fulltext":false,"cited_by_count":2,"citation_normalized_percentile":{"value":0.68730215,"is_in_top_1_percent":false,"is_in_top_10_percent":false},"cited_by_percentile_year":{"min":89,"max":94},"biblio":{"volume":"24","issue":null,"first_page":"273","last_page":"278"},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T12101","display_name":"Advanced Bandit Algorithms Research","score":0.9980000257492065,"subfield":{"id":"https://openalex.org/subfields/1803","display_name":"Management Science and Operations Research"},"field":{"id":"https://openalex.org/fields/18","display_name":"Decision Sciences"},"domain":{"id":"https://openalex.org/domains/2","display_name":"Social Sciences"}},"topics":[{"id":"https://openalex.org/T12101","display_name":"Advanced Bandit Algorithms Research","score":0.9980000257492065,"subfield":{"id":"https://openalex.org/subfields/1803","display_name":"Management Science and Operations Research"},"field":{"id":"https://openalex.org/fields/18","display_name":"Decision Sciences"},"domain":{"id":"https://openalex.org/domains/2","display_name":"Social Sciences"}},{"id":"https://openalex.org/T12303","display_name":"Tensor decomposition and applications","score":0.9871000051498413,"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"}},{"id":"https://openalex.org/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.9168000221252441,"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"}}],"keywords":[{"id":"https://openalex.org/keywords/reinforcement-learning","display_name":"Reinforcement learning","score":0.7975575923919678},{"id":"https://openalex.org/keywords/computer-science","display_name":"Computer science","score":0.7430855631828308},{"id":"https://openalex.org/keywords/scheduling","display_name":"Scheduling (production processes)","score":0.5195696353912354},{"id":"https://openalex.org/keywords/tensor","display_name":"Tensor (intrinsic definition)","score":0.44330787658691406},{"id":"https://openalex.org/keywords/artificial-intelligence","display_name":"Artificial intelligence","score":0.3692062497138977},{"id":"https://openalex.org/keywords/mathematical-optimization","display_name":"Mathematical optimization","score":0.3148512840270996},{"id":"https://openalex.org/keywords/mathematics","display_name":"Mathematics","score":0.09835034608840942}],"concepts":[{"id":"https://openalex.org/C97541855","wikidata":"https://www.wikidata.org/wiki/Q830687","display_name":"Reinforcement learning","level":2,"score":0.7975575923919678},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7430855631828308},{"id":"https://openalex.org/C206729178","wikidata":"https://www.wikidata.org/wiki/Q2271896","display_name":"Scheduling (production processes)","level":2,"score":0.5195696353912354},{"id":"https://openalex.org/C155281189","wikidata":"https://www.wikidata.org/wiki/Q3518150","display_name":"Tensor (intrinsic definition)","level":2,"score":0.44330787658691406},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.3692062497138977},{"id":"https://openalex.org/C126255220","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Mathematical optimization","level":1,"score":0.3148512840270996},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.09835034608840942},{"id":"https://openalex.org/C202444582","wikidata":"https://www.wikidata.org/wiki/Q837863","display_name":"Pure mathematics","level":1,"score":0.0}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.1109/ictai52525.2021.00045","is_oa":false,"landing_page_url":"https://doi.org/10.1109/ictai52525.2021.00045","pdf_url":null,"source":{"id":"https://openalex.org/S4363608052","display_name":"2021 IEEE 33rd International Conference on Tools with Artificial Intelligence (ICTAI)","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"conference"},"license":null,"license_id":null,"version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"2021 IEEE 33rd International Conference on Tools with Artificial Intelligence (ICTAI)","raw_type":"proceedings-article"}],"best_oa_location":null,"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":24,"referenced_works":["https://openalex.org/W2108738385","https://openalex.org/W2168405694","https://openalex.org/W2194775991","https://openalex.org/W2295598076","https://openalex.org/W2804032941","https://openalex.org/W2804500013","https://openalex.org/W2896457183","https://openalex.org/W2919115771","https://openalex.org/W2961619211","https://openalex.org/W2962934715","https://openalex.org/W2963114857","https://openalex.org/W2963684088","https://openalex.org/W3012249773","https://openalex.org/W3096395190","https://openalex.org/W4206530644","https://openalex.org/W4214717370","https://openalex.org/W4295312788","https://openalex.org/W4302296459","https://openalex.org/W6676077707","https://openalex.org/W6685352114","https://openalex.org/W6751349269","https://openalex.org/W6752057402","https://openalex.org/W6766978945","https://openalex.org/W6779728309"],"related_works":["https://openalex.org/W4306904969","https://openalex.org/W2138720691","https://openalex.org/W4362501864","https://openalex.org/W4380318855","https://openalex.org/W2031695474","https://openalex.org/W3084456289","https://openalex.org/W2024136090","https://openalex.org/W4391331176","https://openalex.org/W2586732548","https://openalex.org/W3049728571"],"abstract_inverted_index":{"Efficient":[0],"execution":[1],"is":[2,44],"crucial":[3],"to":[4,12,21,103,160,167],"the":[5,45,82,109,119,179],"successful":[6],"deployment":[7],"of":[8,27,67,72,85,111,127,153,157,175],"deep":[9,53],"learning":[10,78],"models":[11],"real-world":[13],"applications.":[14],"Considerable":[15],"recent":[16,47],"effort":[17],"have":[18],"been":[19,185],"devoted":[20],"computer":[22],"systems":[23],"for":[24,30,51,69],"automatic":[25,70],"discovery":[26],"efficient":[28],"schedules":[29],"executing":[31],"tensor":[32,73,128],"programs":[33,129],"on":[34,39,123],"given":[35,171],"hardware":[36,132],"platforms.":[37,133],"Built":[38],"TVM":[40],"[1],":[41],"Ansor":[42,68],"[2]":[43],"most":[46],"and":[48,87,106,114,130],"state-of-the-art":[49],"framework":[50],"auto-scheduling":[52],"neural":[54],"net":[55],"computation":[56],"pipelines.":[57],"In":[58],"this":[59],"paper,":[60],"we":[61],"present":[62],"Bansor,":[63,122],"an":[64,151,172],"improved":[65],"version":[66],"optimization":[71],"programs,":[74],"using":[75],"bandit-based":[76,101],"reinforcement":[77],"(RL).":[79],"We":[80,98,117],"reexamine":[81],"algorithmic":[83],"procedures":[84],"Ansor,":[86],"identify":[88],"two":[89,131],"selection":[90,113],"sub-problems":[91],"where":[92],"RL":[93],"techniques":[94],"are":[95],"readily":[96],"usable.":[97],"then":[99],"introduce":[100],"algorithms":[102],"balance":[104],"exploration":[105],"exploitation":[107],"during":[108],"processes":[110],"sketch":[112],"task":[115],"scheduling.":[116],"evaluate":[118],"resulting":[120],"algorithm,":[121],"a":[124],"wide":[125],"range":[126],"Experiment":[134],"results":[135,170],"show":[136],"that":[137,181],"Bansor":[138,149],"yields":[139],"significant":[140],"improvement":[141],"over":[142],"Ansor.":[143],"On":[144],"hard":[145],"network":[146],"test":[147],"cases,":[148],"uses":[150],"order":[152],"magnitude":[154],"less":[155],"number":[156,174],"measurement":[158,176],"trails":[159],"attain":[161],"Ansor\u2019s":[162,182],"best":[163],"schedules,":[164],"eventually":[165],"converging":[166],"significantly":[168],"better":[169],"equal":[173],"trials,":[177],"despite":[178],"fact":[180],"performance":[183],"has":[184],"superb.":[186]},"counts_by_year":[{"year":2024,"cited_by_count":1},{"year":2022,"cited_by_count":1}],"updated_date":"2026-07-29T14:22:42.915294","created_date":"2025-10-10T00:00:00"}