{"id":"https://openalex.org/W4396783507","doi":"https://doi.org/10.1109/tc.2024.3398424","title":"TensorMap: A Deep RL-Based Tensor Mapping Framework for Spatial Accelerators","display_name":"TensorMap: A Deep RL-Based Tensor Mapping Framework for Spatial Accelerators","publication_year":2024,"publication_date":"2024-05-09","ids":{"openalex":"https://openalex.org/W4396783507","doi":"https://doi.org/10.1109/tc.2024.3398424"},"language":"en","primary_location":{"id":"doi:10.1109/tc.2024.3398424","is_oa":false,"landing_page_url":"http://dx.doi.org/10.1109/tc.2024.3398424","pdf_url":null,"source":{"id":"https://openalex.org/S157670870","display_name":"IEEE Transactions on Computers","issn_l":"0018-9340","issn":["0018-9340","0016-9340","1557-9956","2326-3814"],"is_oa":false,"is_in_doaj":false,"is_core":true,"host_organization":"https://openalex.org/P4310319808","host_organization_name":"Institute of Electrical and Electronics Engineers","host_organization_lineage":["https://openalex.org/P4310319808"],"host_organization_lineage_names":["Institute of Electrical and Electronics Engineers"],"type":"journal"},"license":null,"license_id":null,"version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"IEEE Transactions on Computers","raw_type":"journal-article"},"type":"article","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/A5101534623","display_name":"Fuyu Wang","orcid":"https://orcid.org/0000-0003-3165-873X"},"institutions":[{"id":"https://openalex.org/I157773358","display_name":"Sun Yat-sen University","ror":"https://ror.org/0064kty71","country_code":"CN","type":"education","lineage":["https://openalex.org/I157773358"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Fuyu Wang","raw_affiliation_strings":["School of Computer Science and Engineering, Sun Yat-Sen University, Guangzhou, China"],"raw_orcid":"https://orcid.org/0000-0003-3165-873X","affiliations":[{"raw_affiliation_string":"School of Computer Science and Engineering, Sun Yat-Sen University, Guangzhou, China","institution_ids":["https://openalex.org/I157773358"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5002840650","display_name":"Minghua Shen","orcid":"https://orcid.org/0000-0003-4747-8020"},"institutions":[{"id":"https://openalex.org/I157773358","display_name":"Sun Yat-sen University","ror":"https://ror.org/0064kty71","country_code":"CN","type":"education","lineage":["https://openalex.org/I157773358"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Minghua Shen","raw_affiliation_strings":["School of Computer Science and Engineering, Sun Yat-Sen University, Guangzhou, China"],"raw_orcid":"https://orcid.org/0000-0003-4747-8020","affiliations":[{"raw_affiliation_string":"School of Computer Science and Engineering, Sun Yat-Sen University, Guangzhou, China","institution_ids":["https://openalex.org/I157773358"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5101633465","display_name":"Yutong Lu","orcid":"https://orcid.org/0000-0001-5315-3375"},"institutions":[{"id":"https://openalex.org/I157773358","display_name":"Sun Yat-sen University","ror":"https://ror.org/0064kty71","country_code":"CN","type":"education","lineage":["https://openalex.org/I157773358"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Yutong Lu","raw_affiliation_strings":["School of Computer Science and Engineering, Sun Yat-Sen University, Guangzhou, China"],"raw_orcid":"https://orcid.org/0000-0001-5315-3375","affiliations":[{"raw_affiliation_string":"School of Computer Science and Engineering, Sun Yat-Sen University, Guangzhou, China","institution_ids":["https://openalex.org/I157773358"]}]},{"author_position":"last","author":{"id":"https://openalex.org/A5023506057","display_name":"Nong Xiao","orcid":"https://orcid.org/0000-0002-2166-977X"},"institutions":[{"id":"https://openalex.org/I157773358","display_name":"Sun Yat-sen University","ror":"https://ror.org/0064kty71","country_code":"CN","type":"education","lineage":["https://openalex.org/I157773358"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Nong Xiao","raw_affiliation_strings":["School of Computer Science and Engineering, Sun Yat-Sen University, Guangzhou, China"],"raw_orcid":"https://orcid.org/0000-0002-2166-977X","affiliations":[{"raw_affiliation_string":"School of Computer Science and Engineering, Sun Yat-Sen University, Guangzhou, China","institution_ids":["https://openalex.org/I157773358"]}]}],"institutions":[],"countries_distinct_count":1,"institutions_distinct_count":1,"corresponding_author_ids":[],"corresponding_institution_ids":["https://openalex.org/I157773358"],"apc_list":{"value":2495,"currency":"USD","value_usd":2495},"apc_paid":null,"fwci":0.6251,"has_fulltext":false,"cited_by_count":2,"citation_normalized_percentile":{"value":0.66134185,"is_in_top_1_percent":false,"is_in_top_10_percent":false},"cited_by_percentile_year":{"min":94,"max":96},"biblio":{"volume":"73","issue":"8","first_page":"1899","last_page":"1912"},"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.9979000091552734,"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.9979000091552734,"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/T10036","display_name":"Advanced Neural Network Applications","score":0.9621000289916992,"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/T11269","display_name":"Algorithms and Data Compression","score":0.9390000104904175,"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/computer-science","display_name":"Computer science","score":0.7161051630973816},{"id":"https://openalex.org/keywords/kernel","display_name":"Kernel (algebra)","score":0.5562771558761597},{"id":"https://openalex.org/keywords/heuristics","display_name":"Heuristics","score":0.5189344882965088},{"id":"https://openalex.org/keywords/tensor","display_name":"Tensor (intrinsic definition)","score":0.49091988801956177},{"id":"https://openalex.org/keywords/computation","display_name":"Computation","score":0.4862607717514038},{"id":"https://openalex.org/keywords/algorithm","display_name":"Algorithm","score":0.38627657294273376},{"id":"https://openalex.org/keywords/artificial-intelligence","display_name":"Artificial intelligence","score":0.3743923306465149},{"id":"https://openalex.org/keywords/theoretical-computer-science","display_name":"Theoretical computer science","score":0.35998398065567017},{"id":"https://openalex.org/keywords/mathematics","display_name":"Mathematics","score":0.1936810314655304}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7161051630973816},{"id":"https://openalex.org/C74193536","wikidata":"https://www.wikidata.org/wiki/Q574844","display_name":"Kernel (algebra)","level":2,"score":0.5562771558761597},{"id":"https://openalex.org/C127705205","wikidata":"https://www.wikidata.org/wiki/Q5748245","display_name":"Heuristics","level":2,"score":0.5189344882965088},{"id":"https://openalex.org/C155281189","wikidata":"https://www.wikidata.org/wiki/Q3518150","display_name":"Tensor (intrinsic definition)","level":2,"score":0.49091988801956177},{"id":"https://openalex.org/C45374587","wikidata":"https://www.wikidata.org/wiki/Q12525525","display_name":"Computation","level":2,"score":0.4862607717514038},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.38627657294273376},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.3743923306465149},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.35998398065567017},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.1936810314655304},{"id":"https://openalex.org/C114614502","wikidata":"https://www.wikidata.org/wiki/Q76592","display_name":"Combinatorics","level":1,"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/C111919701","wikidata":"https://www.wikidata.org/wiki/Q9135","display_name":"Operating system","level":1,"score":0.0}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.1109/tc.2024.3398424","is_oa":false,"landing_page_url":"http://dx.doi.org/10.1109/tc.2024.3398424","pdf_url":null,"source":{"id":"https://openalex.org/S157670870","display_name":"IEEE Transactions on Computers","issn_l":"0018-9340","issn":["0018-9340","0016-9340","1557-9956","2326-3814"],"is_oa":false,"is_in_doaj":false,"is_core":true,"host_organization":"https://openalex.org/P4310319808","host_organization_name":"Institute of Electrical and Electronics Engineers","host_organization_lineage":["https://openalex.org/P4310319808"],"host_organization_lineage_names":["Institute of Electrical and Electronics Engineers"],"type":"journal"},"license":null,"license_id":null,"version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"IEEE Transactions on Computers","raw_type":"journal-article"}],"best_oa_location":null,"sustainable_development_goals":[{"score":0.5699999928474426,"display_name":"Affordable and clean energy","id":"https://metadata.un.org/sdg/7"}],"awards":[{"id":"https://openalex.org/G4536363530","display_name":null,"funder_award_id":"2022ZD0115304","funder_id":"https://openalex.org/F4320335777","funder_display_name":"National Key Research and Development Program of China"},{"id":"https://openalex.org/G5292652112","display_name":null,"funder_award_id":"62072479","funder_id":"https://openalex.org/F4320321001","funder_display_name":"National Natural Science Foundation of China"},{"id":"https://openalex.org/G6651040689","display_name":null,"funder_award_id":"62332021","funder_id":"https://openalex.org/F4320321001","funder_display_name":"National Natural Science Foundation of China"},{"id":"https://openalex.org/G7847425100","display_name":null,"funder_award_id":"2024B1515020011","funder_id":"https://openalex.org/F4320321921","funder_display_name":"Natural Science Foundation of Guangdong Province"}],"funders":[{"id":"https://openalex.org/F4320321001","display_name":"National Natural Science Foundation of China","ror":"https://ror.org/01h0zpd94"},{"id":"https://openalex.org/F4320321921","display_name":"Natural Science Foundation of Guangdong Province","ror":null},{"id":"https://openalex.org/F4320335777","display_name":"National Key Research and Development Program of China","ror":null}],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":45,"referenced_works":["https://openalex.org/W1901129140","https://openalex.org/W2034761517","https://openalex.org/W2067523571","https://openalex.org/W2077143534","https://openalex.org/W2194775991","https://openalex.org/W2295598076","https://openalex.org/W2606722458","https://openalex.org/W2620974420","https://openalex.org/W2736601468","https://openalex.org/W2896457183","https://openalex.org/W2906043559","https://openalex.org/W2940862705","https://openalex.org/W2945146780","https://openalex.org/W2949967139","https://openalex.org/W2963163009","https://openalex.org/W2979644612","https://openalex.org/W2980200167","https://openalex.org/W3012249773","https://openalex.org/W3119880013","https://openalex.org/W3135807226","https://openalex.org/W3153174751","https://openalex.org/W3171752851","https://openalex.org/W3177452048","https://openalex.org/W3192336523","https://openalex.org/W3193212547","https://openalex.org/W3203303453","https://openalex.org/W3205610680","https://openalex.org/W4200566604","https://openalex.org/W4280489237","https://openalex.org/W4281932226","https://openalex.org/W4302296459","https://openalex.org/W4360831810","https://openalex.org/W4385245566","https://openalex.org/W6637151318","https://openalex.org/W6683204974","https://openalex.org/W6692846177","https://openalex.org/W6713134421","https://openalex.org/W6738144653","https://openalex.org/W6741002519","https://openalex.org/W6745935785","https://openalex.org/W6751349269","https://openalex.org/W6752057402","https://openalex.org/W6766978945","https://openalex.org/W6778883912","https://openalex.org/W6779728309"],"related_works":["https://openalex.org/W2280422768","https://openalex.org/W3143197806","https://openalex.org/W4252555497","https://openalex.org/W3121175838","https://openalex.org/W3016293053","https://openalex.org/W1690653314","https://openalex.org/W2401723157","https://openalex.org/W2065055572","https://openalex.org/W2784269775","https://openalex.org/W2952904874"],"abstract_inverted_index":{"The":[0,28],"mapping":[1,15,51,66,80,91],"of":[2,78],"tensor":[3,54,140],"computation":[4],"is":[5,30,87,104,119],"a":[6,46,61,70,100,116],"complex":[7],"and":[8,68,148,159,168,177],"important":[9],"process":[10],"for":[11,53,65,112,134],"spatial":[12,57,113,136],"accelerators.":[13,58,114],"Today's":[14],"works":[16],"depend":[17],"on":[18,56,153,171],"hand-tuned":[19],"kernel":[20],"libraries":[21],"or":[22],"search-based":[23],"heuristics":[24],"from":[25],"human":[26],"experts.":[27],"former":[29],"time-intensive":[31],"while":[32],"the":[33,76,79,94,124,128],"latter":[34],"easily":[35],"leads":[36],"to":[37,74,89,107,121],"sub-optimal":[38],"performance.":[39],"In":[40,127],"this":[41],"paper,":[42],"we":[43,130],"propose":[44,60,99],"TensorMap,":[45],"deep":[47],"reinforcement":[48],"learning":[49],"(RL)-based":[50],"framework":[52],"computations":[55],"We":[59,97],"sequential":[62],"generation":[63],"mode":[64],"optimization":[67],"construct":[69],"coarse-grained":[71],"action":[72],"space":[73],"reduce":[75],"complexity":[77],"search":[81],"space.":[82],"An":[83],"efficient":[84],"policy":[85],"network":[86],"devised":[88],"optimize":[90],"primitives":[92],"in":[93],"RL-based":[95],"search.":[96],"then":[98],"stop":[101],"signal":[102],"that":[103],"sampled":[105],"fromBernoullidistribution":[106],"facilitate":[108],"multi-level":[109],"loop":[110],"unrolling":[111],"Finally,":[115],"genetic":[117],"algorithm":[118],"employed":[120],"further":[122],"refine":[123],"optimized":[125],"mappings.":[126],"experiments,":[129],"demonstrate":[131],"TensorMap's":[132],"ability":[133],"different":[135],"accelerators":[137],"with":[138,156,174],"various":[139],"computations.":[141],"On":[142,162],"TPU,":[143],"TensorMap":[144,164],"provides":[145,165],"2.6$\\times$,":[146],"2.7$\\times$,":[147],"2.4$\\times$better":[149],"energy-delay":[150],"product":[151],"(EDP)":[152],"average":[154,172],"compared":[155,173],"FlexTensor,":[157,175],"Ansor,":[158,176],"AMOS":[160,178],"respectively.":[161,179],"Eyeriss,":[163],"2.1$\\times$,":[166],"1.8$\\times$,":[167],"1.7$\\times$better":[169],"EDP":[170]},"counts_by_year":[{"year":2025,"cited_by_count":2}],"updated_date":"2026-08-28T12:50:07.497085","created_date":"2025-10-10T00:00:00"}