{"id":"https://openalex.org/W7152411621","doi":"https://doi.org/10.48550/arxiv.2604.06260","title":"$S^3$: Stratified Scaling Search for Test-Time in Diffusion Language Models","display_name":"$S^3$: Stratified Scaling Search for Test-Time in Diffusion Language Models","publication_year":2026,"publication_date":"2026-04-07","ids":{"openalex":"https://openalex.org/W7152411621","doi":"https://doi.org/10.48550/arxiv.2604.06260"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2604.06260","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.06260","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.06260","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5133240297","display_name":"Ahsan Bilal","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Bilal, Ahsan","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5027027201","display_name":"Muhammad Ahmed Mohsin","orcid":"https://orcid.org/0009-0005-2766-0345"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Mohsin, Muhammad Ahmed","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133281356","display_name":"Muhammad Umer","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Umer, Muhammad","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5046938499","display_name":"Asad Aali","orcid":"https://orcid.org/0009-0008-2120-5722"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Aali, Asad","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133275616","display_name":"Muhammad Usman Khanzada","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Khanzada, Muhammad Usman","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5039655720","display_name":"Muhammad Usman Rafique","orcid":"https://orcid.org/0000-0001-5504-5482"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Rafique, Muhammad Usman","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133273591","display_name":"Zihao He","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"He, Zihao","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133259934","display_name":"Emily Fox","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Fox, Emily","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5061058579","display_name":"Dean F. Hougen","orcid":"https://orcid.org/0000-0001-5393-1480"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Hougen, Dean F.","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/T10028","display_name":"Topic Modeling","score":0.2596000134944916,"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.2596000134944916,"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/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.17810000479221344,"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/T10201","display_name":"Speech Recognition and Synthesis","score":0.07039999961853027,"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/scaling","display_name":"Scaling","score":0.6759999990463257},{"id":"https://openalex.org/keywords/inference","display_name":"Inference","score":0.5769000053405762},{"id":"https://openalex.org/keywords/sampling","display_name":"Sampling (signal processing)","score":0.5618000030517578},{"id":"https://openalex.org/keywords/schedule","display_name":"Schedule","score":0.45260000228881836},{"id":"https://openalex.org/keywords/process","display_name":"Process (computing)","score":0.43700000643730164},{"id":"https://openalex.org/keywords/language-model","display_name":"Language model","score":0.430400013923645},{"id":"https://openalex.org/keywords/noise-reduction","display_name":"Noise reduction","score":0.42730000615119934},{"id":"https://openalex.org/keywords/diffusion","display_name":"Diffusion","score":0.421999990940094}],"concepts":[{"id":"https://openalex.org/C99844830","wikidata":"https://www.wikidata.org/wiki/Q102441924","display_name":"Scaling","level":2,"score":0.6759999990463257},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6194999814033508},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.5769000053405762},{"id":"https://openalex.org/C140779682","wikidata":"https://www.wikidata.org/wiki/Q210868","display_name":"Sampling (signal processing)","level":3,"score":0.5618000030517578},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.5192000269889832},{"id":"https://openalex.org/C68387754","wikidata":"https://www.wikidata.org/wiki/Q7271585","display_name":"Schedule","level":2,"score":0.45260000228881836},{"id":"https://openalex.org/C98045186","wikidata":"https://www.wikidata.org/wiki/Q205663","display_name":"Process (computing)","level":2,"score":0.43700000643730164},{"id":"https://openalex.org/C137293760","wikidata":"https://www.wikidata.org/wiki/Q3621696","display_name":"Language model","level":2,"score":0.430400013923645},{"id":"https://openalex.org/C163294075","wikidata":"https://www.wikidata.org/wiki/Q581861","display_name":"Noise reduction","level":2,"score":0.42730000615119934},{"id":"https://openalex.org/C69357855","wikidata":"https://www.wikidata.org/wiki/Q163214","display_name":"Diffusion","level":2,"score":0.421999990940094},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.4138999879360199},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.38199999928474426},{"id":"https://openalex.org/C57273362","wikidata":"https://www.wikidata.org/wiki/Q576722","display_name":"Decoding methods","level":2,"score":0.3686000108718872},{"id":"https://openalex.org/C125411270","wikidata":"https://www.wikidata.org/wiki/Q18653","display_name":"Encoding (memory)","level":2,"score":0.3619999885559082},{"id":"https://openalex.org/C111335779","wikidata":"https://www.wikidata.org/wiki/Q3454686","display_name":"Reduction (mathematics)","level":2,"score":0.33739998936653137},{"id":"https://openalex.org/C125583679","wikidata":"https://www.wikidata.org/wiki/Q755673","display_name":"Search algorithm","level":2,"score":0.32899999618530273},{"id":"https://openalex.org/C52740198","wikidata":"https://www.wikidata.org/wiki/Q1539564","display_name":"Importance sampling","level":3,"score":0.3262999951839447},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.30720001459121704},{"id":"https://openalex.org/C99498987","wikidata":"https://www.wikidata.org/wiki/Q2210247","display_name":"Noise (video)","level":3,"score":0.30630001425743103},{"id":"https://openalex.org/C42058472","wikidata":"https://www.wikidata.org/wiki/Q810214","display_name":"Base (topology)","level":2,"score":0.29660001397132874},{"id":"https://openalex.org/C149441793","wikidata":"https://www.wikidata.org/wiki/Q200726","display_name":"Probability distribution","level":2,"score":0.2745000123977661},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.26100000739097595},{"id":"https://openalex.org/C2778755073","wikidata":"https://www.wikidata.org/wiki/Q10858537","display_name":"Scale (ratio)","level":2,"score":0.25929999351501465},{"id":"https://openalex.org/C2780009758","wikidata":"https://www.wikidata.org/wiki/Q6804172","display_name":"Measure (data warehouse)","level":2,"score":0.25040000677108765}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2604.06260","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.06260","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.06260","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.06260","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":[{"score":0.7111934423446655,"display_name":"Quality Education","id":"https://metadata.un.org/sdg/4"}],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Test-time":[0],"scaling":[1,173],"investigates":[2],"whether":[3],"a":[4,54,89,109,168],"fixed":[5],"diffusion":[6,37],"language":[7],"model":[8,122,153],"(DLM)":[9],"can":[10],"generate":[11],"better":[12],"outputs":[13,116],"when":[14],"given":[15],"more":[16],"inference":[17],"compute,":[18],"without":[19],"additional":[20],"training.":[21],"However,":[22],"naive":[23],"best-of-$K$":[24],"sampling":[25,111],"is":[26],"fundamentally":[27],"limited":[28],"because":[29],"it":[30],"repeatedly":[31],"draws":[32],"from":[33],"the":[34,66,73,102,121,142,151],"same":[35],"base":[36],"distribution,":[38],"whose":[39],"high-probability":[40],"regions":[41],"are":[42],"often":[43],"misaligned":[44],"with":[45,88,125],"high-quality":[46],"outputs.":[47],"We":[48],"propose":[49],"$S^3$":[50,81,135],"(Stratified":[51],"Scaling":[52],"Search),":[53],"classical":[55,162],"verifier-guided":[56],"search":[57,103,163],"method":[58],"that":[59,113,134,161],"improves":[60,137],"generation":[61],"by":[62],"reallocating":[63],"compute":[64],"during":[65],"denoising":[67,79,165],"process":[68],"rather":[69],"than":[70],"only":[71],"at":[72],"final":[74],"output":[75],"stage.":[76],"At":[77],"each":[78],"step,":[80],"expands":[82],"multiple":[83],"candidate":[84],"trajectories,":[85],"evaluates":[86],"them":[87],"lightweight":[90],"reference-free":[91],"verifier,":[92],"and":[93,131,154],"selectively":[94],"resamples":[95],"promising":[96],"candidates":[97],"while":[98,117,149],"preserving":[99],"diversity":[100],"within":[101],"frontier.":[104],"This":[105],"procedure":[106],"effectively":[107],"approximates":[108],"reward-tilted":[110],"distribution":[112],"favors":[114],"higher-quality":[115],"remaining":[118],"anchored":[119],"to":[120],"prior.":[123],"Experiments":[124],"LLaDA-8B-Instruct":[126],"on":[127,145],"MATH-500,":[128],"GSM8K,":[129],"ARC-Challenge,":[130],"TruthfulQA":[132],"demonstrate":[133],"consistently":[136],"performance":[138],"across":[139],"benchmarks,":[140],"achieving":[141],"largest":[143],"gains":[144],"mathematical":[146],"reasoning":[147],"tasks":[148],"leaving":[150],"underlying":[152],"decoding":[155],"schedule":[156],"unchanged.":[157],"These":[158],"results":[159],"show":[160],"over":[164],"trajectories":[166],"provides":[167],"practical":[169],"mechanism":[170],"for":[171],"test-time":[172],"in":[174],"DLMs.":[175]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-04-10T00:00:00"}