{"id":"https://openalex.org/W7152470011","doi":"https://doi.org/10.48550/arxiv.2604.06627","title":"DiffuMask: Diffusion Language Model for Token-level Prompt Pruning","display_name":"DiffuMask: Diffusion Language Model for Token-level Prompt Pruning","publication_year":2026,"publication_date":"2026-04-08","ids":{"openalex":"https://openalex.org/W7152470011","doi":"https://doi.org/10.48550/arxiv.2604.06627"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2604.06627","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.06627","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":false,"raw_source_name":null,"raw_type":"article"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://doi.org/10.48550/arxiv.2604.06627","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5035648499","display_name":"Caleb Zheng","orcid":null},"institutions":[],"countries":[],"is_corresponding":true,"raw_author_name":"Zheng, Caleb","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133276334","display_name":"Jyotika Singh","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Singh, Jyotika","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133300777","display_name":"Fang Tu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Tu, Fang","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133292339","display_name":"Weiyi Sun","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Sun, Weiyi","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5110625303","display_name":"Sujeeth Bharadwaj","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Bharadwaj, Sujeeth","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5109466332","display_name":"Yassine Benajiba","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Benajiba, Yassine","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133301758","display_name":"Sujith Ravi","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Ravi, Sujith","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5051370791","display_name":"Eli Shlizerman","orcid":"https://orcid.org/0000-0002-3136-4531"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Shlizerman, Eli","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5133298390","display_name":"Dan Roth","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Roth, Dan","raw_affiliation_strings":[],"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":9,"corresponding_author_ids":["https://openalex.org/A5035648499"],"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/T10028","display_name":"Topic Modeling","score":0.5760999917984009,"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"}},"topics":[{"id":"https://openalex.org/T10028","display_name":"Topic Modeling","score":0.5760999917984009,"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/T11714","display_name":"Multimodal Machine Learning Applications","score":0.0949999988079071,"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/T13702","display_name":"Machine Learning in Healthcare","score":0.0632999986410141,"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/pruning","display_name":"Pruning","score":0.7188000082969666},{"id":"https://openalex.org/keywords/language-model","display_name":"Language model","score":0.6699000000953674},{"id":"https://openalex.org/keywords/masking","display_name":"Masking (illustration)","score":0.5960999727249146},{"id":"https://openalex.org/keywords/context","display_name":"Context (archaeology)","score":0.5899999737739563},{"id":"https://openalex.org/keywords/process","display_name":"Process (computing)","score":0.5006999969482422},{"id":"https://openalex.org/keywords/security-token","display_name":"Security token","score":0.49880000948905945},{"id":"https://openalex.org/keywords/iterative-and-incremental-development","display_name":"Iterative and incremental development","score":0.4348999857902527}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7856000065803528},{"id":"https://openalex.org/C108010975","wikidata":"https://www.wikidata.org/wiki/Q500094","display_name":"Pruning","level":2,"score":0.7188000082969666},{"id":"https://openalex.org/C137293760","wikidata":"https://www.wikidata.org/wiki/Q3621696","display_name":"Language model","level":2,"score":0.6699000000953674},{"id":"https://openalex.org/C2777402240","wikidata":"https://www.wikidata.org/wiki/Q6783436","display_name":"Masking (illustration)","level":2,"score":0.5960999727249146},{"id":"https://openalex.org/C2779343474","wikidata":"https://www.wikidata.org/wiki/Q3109175","display_name":"Context (archaeology)","level":2,"score":0.5899999737739563},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.5755000114440918},{"id":"https://openalex.org/C98045186","wikidata":"https://www.wikidata.org/wiki/Q205663","display_name":"Process (computing)","level":2,"score":0.5006999969482422},{"id":"https://openalex.org/C48145219","wikidata":"https://www.wikidata.org/wiki/Q1335365","display_name":"Security token","level":2,"score":0.49880000948905945},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.4397999942302704},{"id":"https://openalex.org/C143587482","wikidata":"https://www.wikidata.org/wiki/Q1543216","display_name":"Iterative and incremental development","level":2,"score":0.4348999857902527},{"id":"https://openalex.org/C111335779","wikidata":"https://www.wikidata.org/wiki/Q3454686","display_name":"Reduction (mathematics)","level":2,"score":0.4162999987602234},{"id":"https://openalex.org/C183322885","wikidata":"https://www.wikidata.org/wiki/Q17007702","display_name":"Context model","level":3,"score":0.3594000041484833},{"id":"https://openalex.org/C2779982483","wikidata":"https://www.wikidata.org/wiki/Q6094420","display_name":"Iterative refinement","level":2,"score":0.34940001368522644},{"id":"https://openalex.org/C180016635","wikidata":"https://www.wikidata.org/wiki/Q2712821","display_name":"Compression (physics)","level":2,"score":0.3368000090122223},{"id":"https://openalex.org/C78548338","wikidata":"https://www.wikidata.org/wiki/Q2493","display_name":"Data compression","level":2,"score":0.30090001225471497},{"id":"https://openalex.org/C2775924081","wikidata":"https://www.wikidata.org/wiki/Q55608371","display_name":"Control (management)","level":2,"score":0.29820001125335693},{"id":"https://openalex.org/C163294075","wikidata":"https://www.wikidata.org/wiki/Q581861","display_name":"Noise reduction","level":2,"score":0.274399995803833},{"id":"https://openalex.org/C204321447","wikidata":"https://www.wikidata.org/wiki/Q30642","display_name":"Natural language processing","level":1,"score":0.2703000009059906},{"id":"https://openalex.org/C97541855","wikidata":"https://www.wikidata.org/wiki/Q830687","display_name":"Reinforcement learning","level":2,"score":0.2685999870300293}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2604.06627","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.06627","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":"doi:10.48550/arxiv.2604.06627","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.06627","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":false,"raw_source_name":null,"raw_type":"article"},"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":{"In-Context":[0],"Learning":[1],"and":[2,58,65,98,115,126,134],"Chain-of-Thought":[3],"prompting":[4],"improve":[5],"reasoning":[6,96,138],"in":[7,83,139],"large":[8],"language":[9],"models":[10],"(LLMs).":[11],"These":[12],"typically":[13],"come":[14],"at":[15],"the":[16,76],"cost":[17],"of":[18],"longer,":[19],"more":[20,135],"expensive":[21],"prompts":[22],"that":[23,62,121],"may":[24],"contain":[25],"redundant":[26],"information.":[27],"Prompt":[28],"compression":[29,77],"based":[30],"on":[31,41],"pruning":[32,60,68],"offers":[33,88],"a":[34,52,124],"practical":[35],"solution,":[36],"yet":[37],"existing":[38],"methods":[39],"rely":[40],"sequential":[42],"token":[43],"removal":[44],"which":[45],"is":[46],"computationally":[47],"intensive.":[48],"We":[49],"present":[50],"DiffuMask,":[51],"diffusion-based":[53],"framework":[54,128],"integrating":[55],"hierarchical":[56],"shot-level":[57],"token-level":[59],"signals,":[61],"enables":[63],"rapid":[64],"parallel":[66],"prompt":[67,103,130],"via":[69,79],"iterative":[70],"mask":[71],"prediction.":[72],"DiffuMask":[73,122],"substantially":[74],"accelerates":[75],"process":[78],"masking":[80],"multiple":[81],"tokens":[82],"each":[84],"denoising":[85],"step.":[86],"It":[87],"tunable":[89],"control":[90],"over":[91],"retained":[92],"content,":[93],"preserving":[94],"essential":[95],"context":[97],"achieving":[99],"up":[100],"to":[101],"80\\%":[102],"length":[104],"reduction.":[105],"Meanwhile,":[106],"it":[107],"maintains":[108],"or":[109],"improves":[110],"accuracy":[111],"across":[112],"in-domain,":[113],"out-of-domain,":[114],"cross-model":[116],"settings.":[117],"Our":[118],"results":[119],"show":[120],"provides":[123],"generalizable":[125],"controllable":[127],"for":[129],"compression,":[131],"facilitating":[132],"faster":[133],"reliable":[136],"in-context":[137],"LLMs.":[140]},"counts_by_year":[],"updated_date":"2026-04-10T06:07:51.998497","created_date":"2026-04-10T00:00:00"}