{"id":"https://openalex.org/W7154454411","doi":"https://doi.org/10.48550/arxiv.2604.12056","title":"LoSA: Locality Aware Sparse Attention for Block-Wise Diffusion Language Models","display_name":"LoSA: Locality Aware Sparse Attention for Block-Wise Diffusion Language Models","publication_year":2026,"publication_date":"2026-04-13","ids":{"openalex":"https://openalex.org/W7154454411","doi":"https://doi.org/10.48550/arxiv.2604.12056"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2604.12056","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.12056","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":"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.2604.12056","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5133644672","display_name":"Haocheng Xi","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Xi, Haocheng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133380801","display_name":"Harman Singh","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Singh, Harman","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133628986","display_name":"Yuezhou Hu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Hu, Yuezhou","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5007438902","display_name":"Coleman Hooper","orcid":"https://orcid.org/0000-0002-5890-610X"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Hooper, Coleman","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133713327","display_name":"Rishabh Tiwari","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Tiwari, Rishabh","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5093796325","display_name":"Aditya Tomar","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Tomar, Aditya","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133654440","display_name":"Minjae Lee","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Lee, Minjae","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5000868881","display_name":"Wonjun Kang","orcid":"https://orcid.org/0000-0001-5369-6011"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kang, Wonjun","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5127849693","display_name":"Michael Mahoney","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Mahoney, Michael","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133697480","display_name":"Chenfeng Xu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Xu, Chenfeng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133660199","display_name":"Kurt Keutzer","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Keutzer, Kurt","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5133632024","display_name":"Amir Gholami","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Gholami, Amir","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/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.16859999299049377,"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.16859999299049377,"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/T10028","display_name":"Topic Modeling","score":0.13779999315738678,"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/T10201","display_name":"Speech Recognition and Synthesis","score":0.10220000147819519,"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/speedup","display_name":"Speedup","score":0.7152000069618225},{"id":"https://openalex.org/keywords/locality","display_name":"Locality","score":0.508400022983551},{"id":"https://openalex.org/keywords/decoding-methods","display_name":"Decoding methods","score":0.47780001163482666},{"id":"https://openalex.org/keywords/language-model","display_name":"Language model","score":0.4690000116825104},{"id":"https://openalex.org/keywords/fraction","display_name":"Fraction (chemistry)","score":0.39969998598098755},{"id":"https://openalex.org/keywords/prefix","display_name":"Prefix","score":0.3847000002861023},{"id":"https://openalex.org/keywords/diffusion","display_name":"Diffusion","score":0.38089999556541443},{"id":"https://openalex.org/keywords/reduction","display_name":"Reduction (mathematics)","score":0.32899999618530273}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7342000007629395},{"id":"https://openalex.org/C68339613","wikidata":"https://www.wikidata.org/wiki/Q1549489","display_name":"Speedup","level":2,"score":0.7152000069618225},{"id":"https://openalex.org/C2779808786","wikidata":"https://www.wikidata.org/wiki/Q6664603","display_name":"Locality","level":2,"score":0.508400022983551},{"id":"https://openalex.org/C57273362","wikidata":"https://www.wikidata.org/wiki/Q576722","display_name":"Decoding methods","level":2,"score":0.47780001163482666},{"id":"https://openalex.org/C137293760","wikidata":"https://www.wikidata.org/wiki/Q3621696","display_name":"Language model","level":2,"score":0.4690000116825104},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.4081000089645386},{"id":"https://openalex.org/C149629883","wikidata":"https://www.wikidata.org/wiki/Q660926","display_name":"Fraction (chemistry)","level":2,"score":0.39969998598098755},{"id":"https://openalex.org/C141603448","wikidata":"https://www.wikidata.org/wiki/Q134830","display_name":"Prefix","level":2,"score":0.3847000002861023},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.3817000091075897},{"id":"https://openalex.org/C69357855","wikidata":"https://www.wikidata.org/wiki/Q163214","display_name":"Diffusion","level":2,"score":0.38089999556541443},{"id":"https://openalex.org/C111335779","wikidata":"https://www.wikidata.org/wiki/Q3454686","display_name":"Reduction (mathematics)","level":2,"score":0.32899999618530273},{"id":"https://openalex.org/C115537543","wikidata":"https://www.wikidata.org/wiki/Q165596","display_name":"Cache","level":2,"score":0.31850001215934753},{"id":"https://openalex.org/C153180895","wikidata":"https://www.wikidata.org/wiki/Q7148389","display_name":"Pattern recognition (psychology)","level":2,"score":0.3091999888420105},{"id":"https://openalex.org/C159877910","wikidata":"https://www.wikidata.org/wiki/Q2202883","display_name":"Autoregressive model","level":2,"score":0.3012000024318695},{"id":"https://openalex.org/C56372850","wikidata":"https://www.wikidata.org/wiki/Q1050404","display_name":"Sparse matrix","level":3,"score":0.2912999987602234},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.28700000047683716},{"id":"https://openalex.org/C12713177","wikidata":"https://www.wikidata.org/wiki/Q1900281","display_name":"Perspective (graphical)","level":2,"score":0.28119999170303345},{"id":"https://openalex.org/C125411270","wikidata":"https://www.wikidata.org/wiki/Q18653","display_name":"Encoding (memory)","level":2,"score":0.2754000127315521},{"id":"https://openalex.org/C2780009758","wikidata":"https://www.wikidata.org/wiki/Q6804172","display_name":"Measure (data warehouse)","level":2,"score":0.2743000090122223},{"id":"https://openalex.org/C61224824","wikidata":"https://www.wikidata.org/wiki/Q2260434","display_name":"Mixture model","level":2,"score":0.2720000147819519},{"id":"https://openalex.org/C2777601683","wikidata":"https://www.wikidata.org/wiki/Q6499736","display_name":"Vocabulary","level":2,"score":0.2696000039577484},{"id":"https://openalex.org/C206134035","wikidata":"https://www.wikidata.org/wiki/Q811525","display_name":"Treebank","level":3,"score":0.25589999556541443},{"id":"https://openalex.org/C200941418","wikidata":"https://www.wikidata.org/wiki/Q273508","display_name":"Inflation (cosmology)","level":2,"score":0.25450000166893005}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2604.12056","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.12056","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":"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.2604.12056","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.12056","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":"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":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Block-wise":[0],"diffusion":[1],"language":[2],"models":[3],"(DLMs)":[4],"generate":[5],"multiple":[6,134],"tokens":[7,74,84,105],"in":[8,28,152],"any":[9],"order,":[10],"offering":[11],"a":[12,39,69],"promising":[13],"alternative":[14],"to":[15,38,111,149,169],"the":[16,51,80,117,178,181],"autoregressive":[17],"decoding":[18],"pipeline.":[19],"However,":[20],"they":[21],"still":[22],"remain":[23,85],"bottlenecked":[24],"by":[25],"memory-bound":[26],"attention":[27,33,109,163,171],"long-context":[29],"scenarios.":[30],"Naive":[31],"sparse":[32,108],"fails":[34],"on":[35,89,173],"DLMs":[36,136],"due":[37],"KV":[40,55,120],"Inflation":[41],"problem,":[42],"where":[43],"different":[44,47],"queries":[45],"select":[46],"prefix":[48],"positions,":[49],"making":[50],"union":[52],"of":[53,72,82,119,180],"accessed":[54],"pages":[56],"large.":[57],"To":[58],"address":[59],"this,":[60],"we":[61,92],"observe":[62],"that":[63,122],"between":[64],"consecutive":[65],"denoising":[66],"steps,":[67],"only":[68,110],"small":[70],"fraction":[71],"active":[73,112],"exhibit":[75],"significant":[76],"hidden-state":[77],"changes,":[78],"while":[79,143,159],"majority":[81],"stable":[83,104],"nearly":[86],"constant.":[87],"Based":[88],"this":[90],"insight,":[91],"propose":[93],"LOSA":[94,139],"(Locality-aware":[95],"Sparse":[96],"Attention),":[97],"which":[98],"reuses":[99],"cached":[100],"prefix-attention":[101],"results":[102],"for":[103],"and":[106,130,137],"applies":[107],"tokens.":[113],"This":[114],"substantially":[115],"shrinks":[116],"number":[118],"indices":[121],"must":[123],"be":[124],"loaded,":[125],"yielding":[126],"both":[127],"higher":[128,131],"speedup":[129,172],"accuracy.":[132],"Across":[133],"block-wise":[135],"benchmarks,":[138],"preserves":[140],"near-dense":[141],"accuracy":[142,154],"significantly":[144],"improving":[145],"efficiency,":[146],"achieving":[147],"up":[148,168],"+9":[150],"points":[151],"average":[153],"at":[155],"aggressive":[156],"sparsity":[157],"levels":[158],"maintaining":[160],"1.54x":[161],"lower":[162],"density.":[164],"It":[165],"also":[166],"achieves":[167],"4.14x":[170],"RTX":[174],"A6000":[175],"GPUs,":[176],"demonstrating":[177],"effectiveness":[179],"proposed":[182],"method.":[183]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-04-16T00:00:00"}