{"id":"https://openalex.org/W7119082120","doi":"https://doi.org/10.48550/arxiv.2601.03178","title":"DiffBench Meets DiffAgent: End-to-End LLM-Driven Diffusion Acceleration Code Generation","display_name":"DiffBench Meets DiffAgent: End-to-End LLM-Driven Diffusion Acceleration Code Generation","publication_year":2026,"publication_date":"2026-01-06","ids":{"openalex":"https://openalex.org/W7119082120","doi":"https://doi.org/10.48550/arxiv.2601.03178"},"language":"en","primary_location":{"id":"pmh:oai:arXiv.org:2601.03178","doi":null,"is_oa":true,"landing_page_url":"http://arxiv.org/abs/2601.03178","pdf_url":"https://arxiv.org/pdf/2601.03178","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,"listed_in":[],"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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},"volume":null,"type":"preprint","indexed_in":["arxiv","datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://arxiv.org/pdf/2601.03178","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5111333862","display_name":"Jiajun Jiao","orcid":null,"observed_orcids":[]},"institutions":[{"id":"https://openalex.org/I20231570","display_name":"Peking University","ror":"https://ror.org/02v51f717","country_code":"CN","type":"education","lineage":["https://openalex.org/I20231570"]},{"id":"https://openalex.org/I4210137977","display_name":"Advanced Micro Devices (United States)","ror":"https://ror.org/04kd6c783","country_code":"US","type":"company","lineage":["https://openalex.org/I4210137977"]}],"countries":["CN","US"],"is_corresponding":false,"raw_author_name":"jiao, Jiajun","raw_affiliation_strings":["Advanced Micro Devices , Inc","Peking University"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Advanced Micro Devices , Inc","institution_ids":["https://openalex.org/I4210137977"]},{"raw_affiliation_string":"Peking University","institution_ids":["https://openalex.org/I20231570"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5122017786","display_name":"Haowei Zhu","orcid":null,"observed_orcids":[]},"institutions":[{"id":"https://openalex.org/I4210137977","display_name":"Advanced Micro Devices (United States)","ror":"https://ror.org/04kd6c783","country_code":"US","type":"company","lineage":["https://openalex.org/I4210137977"]},{"id":"https://openalex.org/I99065089","display_name":"Tsinghua University","ror":"https://ror.org/03cve4549","country_code":"CN","type":"education","lineage":["https://openalex.org/I99065089"]}],"countries":["CN","US"],"is_corresponding":false,"raw_author_name":"Zhu, Haowei","raw_affiliation_strings":["Advanced Micro Devices , Inc","Tsinghua University {jiajjiao , haoweiz ,"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Advanced Micro Devices , Inc","institution_ids":["https://openalex.org/I4210137977"]},{"raw_affiliation_string":"Tsinghua University {jiajjiao , haoweiz ,","institution_ids":["https://openalex.org/I99065089"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5121954616","display_name":"Puyuan Yang","orcid":null,"observed_orcids":[]},"institutions":[{"id":"https://openalex.org/I4210137977","display_name":"Advanced Micro Devices (United States)","ror":"https://ror.org/04kd6c783","country_code":"US","type":"company","lineage":["https://openalex.org/I4210137977"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Yang, Puyuan","raw_affiliation_strings":["Advanced Micro Devices , Inc"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Advanced Micro Devices , Inc","institution_ids":["https://openalex.org/I4210137977"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5112943670","display_name":"Jianghui Wang","orcid":"https://orcid.org/0009-0006-5884-5868","observed_orcids":["https://orcid.org/0009-0006-5884-5868"]},"institutions":[{"id":"https://openalex.org/I4210137977","display_name":"Advanced Micro Devices (United States)","ror":"https://ror.org/04kd6c783","country_code":"US","type":"company","lineage":["https://openalex.org/I4210137977"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Wang, Jianghui","raw_affiliation_strings":["Advanced Micro Devices , Inc"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Advanced Micro Devices , Inc","institution_ids":["https://openalex.org/I4210137977"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133176699","display_name":"Ji Liu","orcid":null,"observed_orcids":[]},"institutions":[{"id":"https://openalex.org/I4210137977","display_name":"Advanced Micro Devices (United States)","ror":"https://ror.org/04kd6c783","country_code":"US","type":"company","lineage":["https://openalex.org/I4210137977"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Liu, Ji","raw_affiliation_strings":["Advanced Micro Devices , Inc"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Advanced Micro Devices , Inc","institution_ids":["https://openalex.org/I4210137977"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5079691252","display_name":"Ziqiong Liu","orcid":null,"observed_orcids":[]},"institutions":[{"id":"https://openalex.org/I4210137977","display_name":"Advanced Micro Devices (United States)","ror":"https://ror.org/04kd6c783","country_code":"US","type":"company","lineage":["https://openalex.org/I4210137977"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Liu, Ziqiong","raw_affiliation_strings":["Advanced Micro Devices , Inc"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Advanced Micro Devices , Inc","institution_ids":["https://openalex.org/I4210137977"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136079261","display_name":"Dong Li","orcid":null,"observed_orcids":[]},"institutions":[{"id":"https://openalex.org/I4210137977","display_name":"Advanced Micro Devices (United States)","ror":"https://ror.org/04kd6c783","country_code":"US","type":"company","lineage":["https://openalex.org/I4210137977"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Li, Dong","raw_affiliation_strings":["Advanced Micro Devices , Inc"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Advanced Micro Devices , Inc","institution_ids":["https://openalex.org/I4210137977"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5088688674","display_name":"Yuejian Fang","orcid":"https://orcid.org/0000-0002-8279-6908","observed_orcids":["https://orcid.org/0000-0002-8279-6908"]},"institutions":[{"id":"https://openalex.org/I20231570","display_name":"Peking University","ror":"https://ror.org/02v51f717","country_code":"CN","type":"education","lineage":["https://openalex.org/I20231570"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Fang, Yuejian","raw_affiliation_strings":["Peking University"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Peking University","institution_ids":["https://openalex.org/I20231570"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5128142500","display_name":"Jun-hai Yong A","orcid":null,"observed_orcids":[]},"institutions":[{"id":"https://openalex.org/I99065089","display_name":"Tsinghua University","ror":"https://ror.org/03cve4549","country_code":"CN","type":"education","lineage":["https://openalex.org/I99065089"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Yong, Junhai","raw_affiliation_strings":["Tsinghua University {jiajjiao , haoweiz ,"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Tsinghua University {jiajjiao , haoweiz ,","institution_ids":["https://openalex.org/I99065089"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133110275","display_name":"Bin Wang","orcid":null,"observed_orcids":[]},"institutions":[{"id":"https://openalex.org/I99065089","display_name":"Tsinghua University","ror":"https://ror.org/03cve4549","country_code":"CN","type":"education","lineage":["https://openalex.org/I99065089"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Wang, Bin","raw_affiliation_strings":["Tsinghua University {jiajjiao , haoweiz ,"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Tsinghua University {jiajjiao , haoweiz ,","institution_ids":["https://openalex.org/I99065089"]}]},{"author_position":"last","author":{"id":"https://openalex.org/A5027115167","display_name":"Emad Barsoum","orcid":"https://orcid.org/0000-0002-4097-8690","observed_orcids":["https://orcid.org/0000-0002-4097-8690"]},"institutions":[{"id":"https://openalex.org/I4210137977","display_name":"Advanced Micro Devices (United States)","ror":"https://ror.org/04kd6c783","country_code":"US","type":"company","lineage":["https://openalex.org/I4210137977"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Barsoum, Emad","raw_affiliation_strings":["Advanced Micro Devices , Inc"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Advanced Micro Devices , Inc","institution_ids":["https://openalex.org/I4210137977"]}]}],"institutions":[],"countries_distinct_count":2,"institutions_distinct_count":3,"corresponding_author_ids":[],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":null,"has_fulltext":true,"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/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.5205000042915344,"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"}},"topics":[{"id":"https://openalex.org/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.5205000042915344,"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/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.33959999680519104,"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.057999998331069946,"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"}}],"keywords":[{"id":"https://openalex.org/keywords/automated-benchmarking","display_name":"automated benchmarking","score":0.5189999938011169},{"id":"https://openalex.org/keywords/llm-based-agents","display_name":"LLM agents","score":0.5680000185966492},{"id":"https://openalex.org/keywords/genetic-algorithm","display_name":"genetic algorithm","score":0.4269999861717224},{"id":"https://openalex.org/keywords/inference-optimization","display_name":"inference optimization","score":0.3569999933242798},{"id":"https://openalex.org/keywords/image-generation","display_name":"image generation","score":0.4059999883174896},{"id":"https://openalex.org/keywords/video-generation","display_name":"video generation","score":0.734000027179718}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7202000021934509},{"id":"https://openalex.org/C117896860","wikidata":"https://www.wikidata.org/wiki/Q11376","display_name":"Acceleration","level":2,"score":0.7156999707221985},{"id":"https://openalex.org/C168065819","wikidata":"https://www.wikidata.org/wiki/Q845566","display_name":"Debugging","level":2,"score":0.6851000189781189},{"id":"https://openalex.org/C2776760102","wikidata":"https://www.wikidata.org/wiki/Q5139990","display_name":"Code (set theory)","level":3,"score":0.6797000169754028},{"id":"https://openalex.org/C185798385","wikidata":"https://www.wikidata.org/wiki/Q1161707","display_name":"Benchmark (surveying)","level":2,"score":0.6287999749183655},{"id":"https://openalex.org/C168167062","wikidata":"https://www.wikidata.org/wiki/Q1117970","display_name":"Component (thermodynamics)","level":2,"score":0.6190999746322632},{"id":"https://openalex.org/C43521106","wikidata":"https://www.wikidata.org/wiki/Q2165493","display_name":"Pipeline (software)","level":2,"score":0.6054999828338623},{"id":"https://openalex.org/C133162039","wikidata":"https://www.wikidata.org/wiki/Q1061077","display_name":"Code generation","level":3,"score":0.5949000120162964},{"id":"https://openalex.org/C177212765","wikidata":"https://www.wikidata.org/wiki/Q627335","display_name":"Workflow","level":2,"score":0.5501000285148621},{"id":"https://openalex.org/C98045186","wikidata":"https://www.wikidata.org/wiki/Q205663","display_name":"Process (computing)","level":2,"score":0.5422000288963318},{"id":"https://openalex.org/C69357855","wikidata":"https://www.wikidata.org/wiki/Q163214","display_name":"Diffusion","level":2,"score":0.48669999837875366},{"id":"https://openalex.org/C105339364","wikidata":"https://www.wikidata.org/wiki/Q2297740","display_name":"Software deployment","level":2,"score":0.4553999900817871},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.4104999899864197},{"id":"https://openalex.org/C113775141","wikidata":"https://www.wikidata.org/wiki/Q428691","display_name":"Computer engineering","level":1,"score":0.4041000008583069},{"id":"https://openalex.org/C43126263","wikidata":"https://www.wikidata.org/wiki/Q128751","display_name":"Source code","level":2,"score":0.3984000086784363},{"id":"https://openalex.org/C68710425","wikidata":"https://www.wikidata.org/wiki/Q5275442","display_name":"Diffusion process","level":3,"score":0.396699994802475},{"id":"https://openalex.org/C120314980","wikidata":"https://www.wikidata.org/wiki/Q180634","display_name":"Distributed computing","level":1,"score":0.36340001225471497},{"id":"https://openalex.org/C8880873","wikidata":"https://www.wikidata.org/wiki/Q187787","display_name":"Genetic algorithm","level":2,"score":0.3327000141143799},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.31439998745918274},{"id":"https://openalex.org/C2777904410","wikidata":"https://www.wikidata.org/wiki/Q7397","display_name":"Software","level":2,"score":0.29750001430511475},{"id":"https://openalex.org/C125411270","wikidata":"https://www.wikidata.org/wiki/Q18653","display_name":"Encoding (memory)","level":2,"score":0.29420000314712524},{"id":"https://openalex.org/C44154836","wikidata":"https://www.wikidata.org/wiki/Q45045","display_name":"Simulation","level":1,"score":0.28529998660087585},{"id":"https://openalex.org/C128942645","wikidata":"https://www.wikidata.org/wiki/Q1568346","display_name":"Test case","level":3,"score":0.2685999870300293},{"id":"https://openalex.org/C77618280","wikidata":"https://www.wikidata.org/wiki/Q1155772","display_name":"Scheme (mathematics)","level":2,"score":0.26420000195503235},{"id":"https://openalex.org/C79403827","wikidata":"https://www.wikidata.org/wiki/Q3988","display_name":"Real-time computing","level":1,"score":0.26339998841285706},{"id":"https://openalex.org/C34146451","wikidata":"https://www.wikidata.org/wiki/Q5048094","display_name":"Cascade","level":2,"score":0.2614000141620636},{"id":"https://openalex.org/C459310","wikidata":"https://www.wikidata.org/wiki/Q117801","display_name":"Computational science","level":1,"score":0.2565000057220459},{"id":"https://openalex.org/C175309249","wikidata":"https://www.wikidata.org/wiki/Q725864","display_name":"Pipeline transport","level":2,"score":0.2538999915122986}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:oai:arXiv.org:2601.03178","doi":null,"is_oa":true,"landing_page_url":"http://arxiv.org/abs/2601.03178","pdf_url":"https://arxiv.org/pdf/2601.03178","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,"listed_in":[],"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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},{"id":"doi:10.48550/arxiv.2601.03178","doi":"https://doi.org/10.48550/arxiv.2601.03178","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2601.03178","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,"listed_in":[],"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"}],"best_oa_location":{"id":"pmh:oai:arXiv.org:2601.03178","doi":null,"is_oa":true,"landing_page_url":"http://arxiv.org/abs/2601.03178","pdf_url":"https://arxiv.org/pdf/2601.03178","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,"listed_in":[],"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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},"sustainable_development_goals":[],"study_designs":[],"awards":[],"funders":[],"has_content":{"grobid_xml":true,"pdf":true},"content_urls":{"pdf":"https://content.openalex.org/works/W7119082120.pdf","grobid_xml":"https://content.openalex.org/works/W7119082120.grobid-xml"},"referenced_works_count":0,"referenced_works":[],"related_works":["https://openalex.org/W4406273065","https://openalex.org/W7168340407","https://openalex.org/W4415164199","https://openalex.org/W7133329278","https://openalex.org/W4415088691","https://openalex.org/W4416872067","https://openalex.org/W4409366321","https://openalex.org/W4404008794","https://openalex.org/W4415088061","https://openalex.org/W4417102600"],"abstract_inverted_index":{"Diffusion":[0],"models":[1,27,56],"have":[2],"achieved":[3],"remarkable":[4],"success":[5],"in":[6,111,179],"image":[7],"and":[8,63,86,100,116,154,172],"video":[9],"generation.":[10],"However,":[11],"their":[12],"inherently":[13],"multiple":[14,36],"step":[15],"inference":[16],"process":[17],"imposes":[18],"substantial":[19],"computational":[20],"overhead,":[21],"hindering":[22],"real-world":[23],"deployment.":[24],"Accelerating":[25],"diffusion":[26,82,104,182],"is":[28],"therefore":[29],"essential,":[30],"yet":[31],"determining":[32],"how":[33],"to":[34,140],"combine":[35],"model":[37],"acceleration":[38,60,98,183],"techniques":[39],"remains":[40],"a":[41,50,69,74,108,113,117,125,130,147,166],"significant":[42],"challenge.":[43],"To":[44],"address":[45],"this":[46],"issue,":[47],"we":[48,66,90],"introduce":[49],"framework":[51],"driven":[52],"by":[53],"large":[54],"language":[55],"(LLMs)":[57],"for":[58,102],"automated":[59,77],"code":[61,126,143],"generation":[62,127],"evaluation.":[64],"First,":[65],"present":[67],"DiffBench,":[68],"comprehensive":[70],"benchmark":[71],"that":[72,95,163,173],"implements":[73],"three":[75],"stage":[76],"evaluation":[78,168],"pipeline":[79],"across":[80],"diverse":[81],"architectures,":[83],"optimization":[84],"combinations":[85],"deployment":[87],"scenarios.":[88],"Second,":[89],"propose":[91],"DiffAgent,":[92],"an":[93],"agent":[94],"generates":[96],"optimal":[97],"strategies":[99],"codes":[101,171],"arbitrary":[103],"models.":[105],"DiffAgent":[106,174],"employs":[107],"closed-loop":[109],"workflow":[110],"which":[112],"planning":[114],"component":[115,119],"debugging":[118],"iteratively":[120],"refine":[121],"the":[122,137,151,155],"output":[123],"of":[124,150,169],"component,":[128],"while":[129],"genetic":[131],"algorithm":[132],"extracts":[133],"performance":[134],"feedback":[135],"from":[136],"execution":[138],"environment":[139],"guide":[141],"subsequent":[142],"refinements.":[144],"We":[145],"provide":[146],"detailed":[148],"explanation":[149],"DiffBench":[152,164],"construction":[153],"design":[156],"principles":[157],"underlying":[158],"DiffAgent.":[159],"Extensive":[160],"experiments":[161],"show":[162],"offers":[165],"thorough":[167],"generated":[170],"significantly":[175],"outperforms":[176],"existing":[177],"LLMs":[178],"producing":[180],"effective":[181],"strategies.":[184]},"counts_by_year":[],"updated_date":"2026-10-31T12:09:11.096373","created_date":"2026-01-08T00:00:00"}