{"id":"https://openalex.org/W7164877636","doi":"https://doi.org/10.48550/arxiv.2606.14971","title":"FastMix: Fast Data Mixture Optimization via Gradient Descent","display_name":"FastMix: Fast Data Mixture Optimization via Gradient Descent","publication_year":2026,"publication_date":"2026-06-12","ids":{"openalex":"https://openalex.org/W7164877636","doi":"https://doi.org/10.48550/arxiv.2606.14971"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2606.14971","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.14971","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.2606.14971","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5054734725","display_name":"Haoru Tan","orcid":"https://orcid.org/0009-0001-6721-2468"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Tan, Haoru","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5019272206","display_name":"S S Wu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wu, Sitong","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136737796","display_name":"Yanfeng Chen","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Chen, Yanfeng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5138729359","display_name":"Jun Xia","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Xia, Jun","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5138758038","display_name":"Ruobing Xie","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Xie, Ruobing","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5138701106","display_name":"Bin Xia","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Xia, Bin","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5138743754","display_name":"Xingwu Sun","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Sun, Xingwu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5138756555","display_name":"Xiaojuan Qi","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Qi, Xiaojuan","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/T11307","display_name":"Domain Adaptation and Few-Shot Learning","score":0.37059998512268066,"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/T11307","display_name":"Domain Adaptation and Few-Shot Learning","score":0.37059998512268066,"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/T12814","display_name":"Gaussian Processes and Bayesian Inference","score":0.09160000085830688,"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.0738999992609024,"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/mixture-model","display_name":"Mixture model","score":0.7809000015258789},{"id":"https://openalex.org/keywords/bilevel-optimization","display_name":"Bilevel optimization","score":0.5467000007629395},{"id":"https://openalex.org/keywords/gradient-descent","display_name":"Gradient descent","score":0.5209000110626221},{"id":"https://openalex.org/keywords/scalability","display_name":"Scalability","score":0.5084999799728394},{"id":"https://openalex.org/keywords/selection","display_name":"Selection (genetic algorithm)","score":0.46639999747276306},{"id":"https://openalex.org/keywords/convergence","display_name":"Convergence (economics)","score":0.42100000381469727},{"id":"https://openalex.org/keywords/heuristics","display_name":"Heuristics","score":0.41920000314712524},{"id":"https://openalex.org/keywords/mixture-theory","display_name":"Mixture theory","score":0.3919999897480011},{"id":"https://openalex.org/keywords/optimization-problem","display_name":"Optimization problem","score":0.38999998569488525}],"concepts":[{"id":"https://openalex.org/C61224824","wikidata":"https://www.wikidata.org/wiki/Q2260434","display_name":"Mixture model","level":2,"score":0.7809000015258789},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6083999872207642},{"id":"https://openalex.org/C3309286","wikidata":"https://www.wikidata.org/wiki/Q4907693","display_name":"Bilevel optimization","level":3,"score":0.5467000007629395},{"id":"https://openalex.org/C153258448","wikidata":"https://www.wikidata.org/wiki/Q1199743","display_name":"Gradient descent","level":3,"score":0.5209000110626221},{"id":"https://openalex.org/C48044578","wikidata":"https://www.wikidata.org/wiki/Q727490","display_name":"Scalability","level":2,"score":0.5084999799728394},{"id":"https://openalex.org/C126255220","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Mathematical optimization","level":1,"score":0.4948999881744385},{"id":"https://openalex.org/C81917197","wikidata":"https://www.wikidata.org/wiki/Q628760","display_name":"Selection (genetic algorithm)","level":2,"score":0.46639999747276306},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.46209999918937683},{"id":"https://openalex.org/C2777303404","wikidata":"https://www.wikidata.org/wiki/Q759757","display_name":"Convergence (economics)","level":2,"score":0.42100000381469727},{"id":"https://openalex.org/C127705205","wikidata":"https://www.wikidata.org/wiki/Q5748245","display_name":"Heuristics","level":2,"score":0.41920000314712524},{"id":"https://openalex.org/C2779736610","wikidata":"https://www.wikidata.org/wiki/Q6884140","display_name":"Mixture theory","level":3,"score":0.3919999897480011},{"id":"https://openalex.org/C137836250","wikidata":"https://www.wikidata.org/wiki/Q984063","display_name":"Optimization problem","level":2,"score":0.38999998569488525},{"id":"https://openalex.org/C159694833","wikidata":"https://www.wikidata.org/wiki/Q2321565","display_name":"Iterative method","level":2,"score":0.37779998779296875},{"id":"https://openalex.org/C202615002","wikidata":"https://www.wikidata.org/wiki/Q783507","display_name":"Differentiable function","level":2,"score":0.3749000132083893},{"id":"https://openalex.org/C115680565","wikidata":"https://www.wikidata.org/wiki/Q5977448","display_name":"Gradient method","level":2,"score":0.3463999927043915},{"id":"https://openalex.org/C160920958","wikidata":"https://www.wikidata.org/wiki/Q7662746","display_name":"Synthetic data","level":2,"score":0.32109999656677246},{"id":"https://openalex.org/C55660270","wikidata":"https://www.wikidata.org/wiki/Q5164377","display_name":"Constrained optimization","level":2,"score":0.319599986076355},{"id":"https://openalex.org/C93959086","wikidata":"https://www.wikidata.org/wiki/Q6888345","display_name":"Model selection","level":2,"score":0.310699999332428},{"id":"https://openalex.org/C147764199","wikidata":"https://www.wikidata.org/wiki/Q6865248","display_name":"Minification","level":2,"score":0.2872999906539917},{"id":"https://openalex.org/C157553263","wikidata":"https://www.wikidata.org/wiki/Q5168004","display_name":"Coordinate descent","level":2,"score":0.28700000047683716},{"id":"https://openalex.org/C21080849","wikidata":"https://www.wikidata.org/wiki/Q13611879","display_name":"Data point","level":2,"score":0.2865000069141388},{"id":"https://openalex.org/C177179195","wikidata":"https://www.wikidata.org/wiki/Q7268372","display_name":"Quadratic unconstrained binary optimization","level":4,"score":0.2847999930381775},{"id":"https://openalex.org/C12713177","wikidata":"https://www.wikidata.org/wiki/Q1900281","display_name":"Perspective (graphical)","level":2,"score":0.2791999876499176},{"id":"https://openalex.org/C177148314","wikidata":"https://www.wikidata.org/wiki/Q170084","display_name":"Generalization","level":2,"score":0.2786000072956085},{"id":"https://openalex.org/C2164484","wikidata":"https://www.wikidata.org/wiki/Q5170150","display_name":"Core (optical fiber)","level":2,"score":0.2782999873161316},{"id":"https://openalex.org/C43126263","wikidata":"https://www.wikidata.org/wiki/Q128751","display_name":"Source code","level":2,"score":0.27489998936653137},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.2687000036239624},{"id":"https://openalex.org/C148483581","wikidata":"https://www.wikidata.org/wiki/Q446488","display_name":"Feature selection","level":2,"score":0.26809999346733093},{"id":"https://openalex.org/C124101348","wikidata":"https://www.wikidata.org/wiki/Q172491","display_name":"Data mining","level":1,"score":0.26019999384880066},{"id":"https://openalex.org/C126980161","wikidata":"https://www.wikidata.org/wiki/Q863783","display_name":"Simulated annealing","level":2,"score":0.2531999945640564},{"id":"https://openalex.org/C45374587","wikidata":"https://www.wikidata.org/wiki/Q12525525","display_name":"Computation","level":2,"score":0.25200000405311584}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2606.14971","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.14971","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.2606.14971","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.14971","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":[{"id":"https://metadata.un.org/sdg/8","display_name":"Decent work and economic growth","score":0.5233250856399536}],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"While":[0],"large":[1,10],"and":[2,19,61,67,128,158,171],"diverse":[3],"datasets":[4],"have":[5],"driven":[6],"recent":[7],"advances":[8],"in":[9],"models,":[11],"identifying":[12],"the":[13,73,111,116,132],"optimal":[14],"data":[15,37,149],"mixture":[16,38,59,81,95,112,127,154,161],"for":[17],"pre-training":[18],"post-training":[20],"remains":[21],"a":[22,32,43,78,84],"significant":[23],"open":[24],"problem.":[25,87],"We":[26],"address":[27],"this":[28,89],"challenge":[29],"with":[30],"FASTMIX,":[31],"novel":[33],"framework":[34],"that":[35,93],"automates":[36],"discovery":[39],"while":[40,176],"training":[41],"only":[42],"single":[44],"proxy":[45],"model.":[46,129],"Instead":[47],"of":[48,75,80,125],"relying":[49],"on":[50,148,164],"predefined":[51],"heuristics":[52],"or":[53],"resource-intensive":[54],"simulations,":[55],"FASTMIX":[56,76,135,173],"jointly":[57],"optimizes":[58],"coefficients":[60,113],"model":[62,146],"parameters,":[63],"substantially":[64],"improving":[65],"efficiency":[66],"scalability":[68],"over":[69],"prior":[70],"approaches.":[71],"At":[72],"core":[74],"is":[77,97],"reformulation":[79],"selection":[82],"as":[83],"bilevel":[85],"optimization":[86,119,124,133,140],"Under":[88],"reformulation,":[90],"we":[91],"show":[92],"optimizing":[94],"ratios":[96,155,162],"mathematically":[98],"equivalent":[99],"to":[100,152],"assigning":[101],"per-source":[102],"loss":[103],"weights":[104],"under":[105],"uniform":[106],"source":[107],"sampling.":[108],"This":[109],"embeds":[110],"directly":[114],"into":[115],"differentiable":[117],"iterative":[118,139],"objective,":[120],"enabling":[121],"efficient,":[122],"gradient-based":[123],"both":[126],"To":[130],"solve":[131],"problem,":[134],"implements":[136],"an":[137],"approximate":[138],"procedure,":[141],"alternating":[142],"between":[143],"(i)":[144],"updating":[145,160],"parameters":[147],"sampled":[150],"according":[151],"current":[153],"(inner":[156],"loop)":[157],"(ii)":[159],"based":[163],"validation":[165],"feedback":[166],"(outer":[167],"loop).":[168],"Across":[169],"pre-":[170],"post-training,":[172],"outperforms":[174],"baselines":[175],"drastically":[177],"reducing":[178],"search":[179],"cost.":[180],"Code":[181],"(https://github.com/hrtan/fastmix)":[182]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-06-17T00:00:00"}