{"id":"https://openalex.org/W7154251735","doi":"https://doi.org/10.48550/arxiv.2604.11351","title":"WM-DAgger: Enabling Efficient Data Aggregation for Imitation Learning with World Models","display_name":"WM-DAgger: Enabling Efficient Data Aggregation for Imitation Learning with World Models","publication_year":2026,"publication_date":"2026-04-13","ids":{"openalex":"https://openalex.org/W7154251735","doi":"https://doi.org/10.48550/arxiv.2604.11351"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2604.11351","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.11351","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":"Preprint"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://doi.org/10.48550/arxiv.2604.11351","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5133583367","display_name":"Anlan Yu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yu, Anlan","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133577865","display_name":"Zaishu Chen","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Chen, Zaishu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133571759","display_name":"Peili Song","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Song, Peili","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5077830473","display_name":"Zhiqing Hong","orcid":"https://orcid.org/0000-0003-3682-4290"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Hong, Zhiqing","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133608161","display_name":"Haotian Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Haotian","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5100603765","display_name":"Desheng Zhang","orcid":"https://orcid.org/0009-0004-3986-6394"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Desheng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133594447","display_name":"Tian He","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"He, Tian","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133600853","display_name":"Yi Ding","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Ding, Yi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5123809212","display_name":"Daqing Zhang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Daqing","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/T10653","display_name":"Robot Manipulation and Learning","score":0.5651000142097473,"subfield":{"id":"https://openalex.org/subfields/2207","display_name":"Control and Systems Engineering"},"field":{"id":"https://openalex.org/fields/22","display_name":"Engineering"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},"topics":[{"id":"https://openalex.org/T10653","display_name":"Robot Manipulation and Learning","score":0.5651000142097473,"subfield":{"id":"https://openalex.org/subfields/2207","display_name":"Control and Systems Engineering"},"field":{"id":"https://openalex.org/fields/22","display_name":"Engineering"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10462","display_name":"Reinforcement Learning in Robotics","score":0.2549999952316284,"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/T10709","display_name":"Social Robot Interaction and HRI","score":0.043800000101327896,"subfield":{"id":"https://openalex.org/subfields/3207","display_name":"Social Psychology"},"field":{"id":"https://openalex.org/fields/32","display_name":"Psychology"},"domain":{"id":"https://openalex.org/domains/2","display_name":"Social Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/key","display_name":"Key (lock)","score":0.617900013923645},{"id":"https://openalex.org/keywords/focus","display_name":"Focus (optics)","score":0.5526999831199646},{"id":"https://openalex.org/keywords/imitation","display_name":"Imitation","score":0.5257999897003174},{"id":"https://openalex.org/keywords/robot","display_name":"Robot","score":0.48019999265670776},{"id":"https://openalex.org/keywords/action","display_name":"Action (physics)","score":0.4749999940395355},{"id":"https://openalex.org/keywords/frame","display_name":"Frame (networking)","score":0.4101000130176544},{"id":"https://openalex.org/keywords/code","display_name":"Code (set theory)","score":0.40290001034736633},{"id":"https://openalex.org/keywords/training-set","display_name":"Training set","score":0.3831000030040741}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7454000115394592},{"id":"https://openalex.org/C26517878","wikidata":"https://www.wikidata.org/wiki/Q228039","display_name":"Key (lock)","level":2,"score":0.617900013923645},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.593999981880188},{"id":"https://openalex.org/C192209626","wikidata":"https://www.wikidata.org/wiki/Q190909","display_name":"Focus (optics)","level":2,"score":0.5526999831199646},{"id":"https://openalex.org/C126388530","wikidata":"https://www.wikidata.org/wiki/Q1131737","display_name":"Imitation","level":2,"score":0.5257999897003174},{"id":"https://openalex.org/C90509273","wikidata":"https://www.wikidata.org/wiki/Q11012","display_name":"Robot","level":2,"score":0.48019999265670776},{"id":"https://openalex.org/C2780791683","wikidata":"https://www.wikidata.org/wiki/Q846785","display_name":"Action (physics)","level":2,"score":0.4749999940395355},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.4431000053882599},{"id":"https://openalex.org/C126042441","wikidata":"https://www.wikidata.org/wiki/Q1324888","display_name":"Frame (networking)","level":2,"score":0.4101000130176544},{"id":"https://openalex.org/C2776760102","wikidata":"https://www.wikidata.org/wiki/Q5139990","display_name":"Code (set theory)","level":3,"score":0.40290001034736633},{"id":"https://openalex.org/C51632099","wikidata":"https://www.wikidata.org/wiki/Q3985153","display_name":"Training set","level":2,"score":0.3831000030040741},{"id":"https://openalex.org/C97541855","wikidata":"https://www.wikidata.org/wiki/Q830687","display_name":"Reinforcement learning","level":2,"score":0.3571999967098236},{"id":"https://openalex.org/C2779038628","wikidata":"https://www.wikidata.org/wiki/Q7248497","display_name":"Programming by demonstration","level":3,"score":0.3465000092983246},{"id":"https://openalex.org/C107457646","wikidata":"https://www.wikidata.org/wiki/Q207434","display_name":"Human\u2013computer interaction","level":1,"score":0.33469998836517334},{"id":"https://openalex.org/C67186912","wikidata":"https://www.wikidata.org/wiki/Q367664","display_name":"Data modeling","level":2,"score":0.30809998512268066},{"id":"https://openalex.org/C82578977","wikidata":"https://www.wikidata.org/wiki/Q16773055","display_name":"Data aggregator","level":3,"score":0.3021000027656555},{"id":"https://openalex.org/C2776145971","wikidata":"https://www.wikidata.org/wiki/Q30673951","display_name":"Labeled data","level":2,"score":0.2969000041484833},{"id":"https://openalex.org/C13662910","wikidata":"https://www.wikidata.org/wiki/Q193139","display_name":"Trajectory","level":2,"score":0.27469998598098755},{"id":"https://openalex.org/C774472","wikidata":"https://www.wikidata.org/wiki/Q6760393","display_name":"Margin (machine learning)","level":2,"score":0.2734000086784363},{"id":"https://openalex.org/C145460709","wikidata":"https://www.wikidata.org/wiki/Q859951","display_name":"Human\u2013robot interaction","level":3,"score":0.27219998836517334},{"id":"https://openalex.org/C3020493868","wikidata":"https://www.wikidata.org/wiki/Q55631277","display_name":"Real world data","level":2,"score":0.2700999975204468},{"id":"https://openalex.org/C60692881","wikidata":"https://www.wikidata.org/wiki/Q584529","display_name":"Humanoid robot","level":3,"score":0.26339998841285706}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2604.11351","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.11351","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":"Preprint"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2604.11351","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.11351","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":"Preprint"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Imitation":[0],"learning":[1],"is":[2,13,195],"a":[3,125,141,180],"powerful":[4],"paradigm":[5],"for":[6],"training":[7,31],"robotic":[8,98,169],"policies,":[9],"yet":[10],"its":[11,56],"performance":[12],"limited":[14],"by":[15,150],"compounding":[16],"errors:":[17],"minor":[18],"policy":[19,35],"inaccuracies":[20],"could":[21,36],"drive":[22],"robots":[23],"into":[24],"unseen":[25],"out-of-distribution":[26],"(OOD)":[27],"states":[28],"in":[29,159,184],"the":[30,34,46,111],"set,":[32],"where":[33],"generate":[37],"even":[38],"bigger":[39],"errors,":[40],"leading":[41],"to":[42,52,80,115,135,155],"eventual":[43],"failures.":[44],"While":[45],"Data":[47],"Aggregation":[48],"(DAgger)":[49],"framework":[50,75,119],"tries":[51],"address":[53],"this":[54,66],"issue,":[55],"reliance":[57],"on":[58,92,166],"continuous":[59],"human":[60,87],"involvement":[61],"severely":[62],"limits":[63],"scalability.":[64],"In":[65],"paper,":[67],"we":[68,90],"propose":[69],"WM-DAgger,":[70],"an":[71,96],"efficient":[72],"data":[73,84,108],"aggregation":[74],"that":[76,130,145,172],"leverages":[77],"World":[78,116],"Models":[79],"synthesize":[81],"OOD":[82],"recovery":[83,133],"without":[85],"requiring":[86],"involvement.":[88],"Specifically,":[89],"focus":[91],"manipulation":[93],"tasks":[94],"with":[95,188],"eye-in-hand":[97],"arm":[99],"and":[100,109,139],"only":[101,189],"few-shot":[102],"demonstrations.":[103,161,191],"To":[104],"avoid":[105],"synthesizing":[106],"misleading":[107,137],"overcome":[110],"hallucination":[112],"issues":[113],"inherent":[114],"Models,":[117],"our":[118,173],"introduces":[120],"two":[121],"key":[122],"mechanisms:":[123],"(1)":[124],"Corrective":[126],"Action":[127],"Synthesis":[128],"Module":[129,144],"generates":[131],"task-oriented":[132],"actions":[134],"prevent":[136],"supervision,":[138],"(2)":[140],"Consistency-Guided":[142],"Filtering":[143],"discards":[146],"physically":[147],"implausible":[148],"trajectories":[149],"anchoring":[151],"terminal":[152],"synthesized":[153],"frames":[154,158],"corresponding":[156],"real":[157],"expert":[160],"We":[162],"extensively":[163],"validate":[164],"WM-DAgger":[165],"multiple":[167],"real-world":[168],"tasks.":[170],"Results":[171],"method":[174],"significantly":[175],"improves":[176],"success":[177,182],"rates,":[178],"achieving":[179],"93.3\\%":[181],"rate":[183],"soft":[185],"bag":[186],"pushing":[187],"five":[190],"The":[192],"source":[193],"code":[194],"publicly":[196],"available":[197],"at":[198],"https://github.com/czs12354-xxdbd/WM-Dagger.":[199]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-04-15T00:00:00"}