{"id":"https://openalex.org/W7133539277","doi":"https://doi.org/10.48550/arxiv.2603.03205","title":"Learning When to Act or Refuse: Guarding Agentic Reasoning Models for Safe Multi-Step Tool Use","display_name":"Learning When to Act or Refuse: Guarding Agentic Reasoning Models for Safe Multi-Step Tool Use","publication_year":2026,"publication_date":"2026-03-03","ids":{"openalex":"https://openalex.org/W7133539277","doi":"https://doi.org/10.48550/arxiv.2603.03205"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2603.03205","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.03205","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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.2603.03205","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5111335327","display_name":"Aradhye Agarwal","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Agarwal, Aradhye","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5128070363","display_name":"Gurdit Siyan","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Siyan, Gurdit","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5119934695","display_name":"Yash Pandya","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Pandya, Yash","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5101278709","display_name":"Joykirat Singh","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Singh, Joykirat","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5128124771","display_name":"Akshay Nambi","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Nambi, Akshay","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5128077209","display_name":"Ahmed M.A. AwadAllah","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Awadallah, Ahmed","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/T11689","display_name":"Adversarial Robustness in Machine Learning","score":0.4754999876022339,"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/T11689","display_name":"Adversarial Robustness in Machine Learning","score":0.4754999876022339,"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/T10028","display_name":"Topic Modeling","score":0.11840000003576279,"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/T12026","display_name":"Explainable Artificial Intelligence (XAI)","score":0.08070000261068344,"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/adversarial-system","display_name":"Adversarial system","score":0.6712999939918518},{"id":"https://openalex.org/keywords/task","display_name":"Task (project management)","score":0.5802000164985657},{"id":"https://openalex.org/keywords/generalization","display_name":"Generalization","score":0.5144000053405762},{"id":"https://openalex.org/keywords/inference","display_name":"Inference","score":0.49540001153945923},{"id":"https://openalex.org/keywords/pairwise-comparison","display_name":"Pairwise comparison","score":0.43380001187324524},{"id":"https://openalex.org/keywords/task-analysis","display_name":"Task analysis","score":0.3553999960422516},{"id":"https://openalex.org/keywords/language-model","display_name":"Language model","score":0.34200000762939453},{"id":"https://openalex.org/keywords/reinforcement-learning","display_name":"Reinforcement learning","score":0.32850000262260437}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6996999979019165},{"id":"https://openalex.org/C37736160","wikidata":"https://www.wikidata.org/wiki/Q1801315","display_name":"Adversarial system","level":2,"score":0.6712999939918518},{"id":"https://openalex.org/C2780451532","wikidata":"https://www.wikidata.org/wiki/Q759676","display_name":"Task (project management)","level":2,"score":0.5802000164985657},{"id":"https://openalex.org/C177148314","wikidata":"https://www.wikidata.org/wiki/Q170084","display_name":"Generalization","level":2,"score":0.5144000053405762},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.49540001153945923},{"id":"https://openalex.org/C107457646","wikidata":"https://www.wikidata.org/wiki/Q207434","display_name":"Human\u2013computer interaction","level":1,"score":0.4596000015735626},{"id":"https://openalex.org/C38652104","wikidata":"https://www.wikidata.org/wiki/Q3510521","display_name":"Computer security","level":1,"score":0.4424000084400177},{"id":"https://openalex.org/C184898388","wikidata":"https://www.wikidata.org/wiki/Q1435712","display_name":"Pairwise comparison","level":2,"score":0.43380001187324524},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.41280001401901245},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.39989998936653137},{"id":"https://openalex.org/C175154964","wikidata":"https://www.wikidata.org/wiki/Q380077","display_name":"Task analysis","level":3,"score":0.3553999960422516},{"id":"https://openalex.org/C137293760","wikidata":"https://www.wikidata.org/wiki/Q3621696","display_name":"Language model","level":2,"score":0.34200000762939453},{"id":"https://openalex.org/C97541855","wikidata":"https://www.wikidata.org/wiki/Q830687","display_name":"Reinforcement learning","level":2,"score":0.32850000262260437},{"id":"https://openalex.org/C48145219","wikidata":"https://www.wikidata.org/wiki/Q1335365","display_name":"Security token","level":2,"score":0.32359999418258667},{"id":"https://openalex.org/C13662910","wikidata":"https://www.wikidata.org/wiki/Q193139","display_name":"Trajectory","level":2,"score":0.2935999929904938},{"id":"https://openalex.org/C84945661","wikidata":"https://www.wikidata.org/wiki/Q7366567","display_name":"Root cause","level":2,"score":0.2847000062465668},{"id":"https://openalex.org/C112930515","wikidata":"https://www.wikidata.org/wiki/Q4389547","display_name":"Risk analysis (engineering)","level":1,"score":0.28200000524520874},{"id":"https://openalex.org/C98184364","wikidata":"https://www.wikidata.org/wiki/Q1780131","display_name":"Argument (complex analysis)","level":2,"score":0.272599995136261},{"id":"https://openalex.org/C180747234","wikidata":"https://www.wikidata.org/wiki/Q23373","display_name":"Cognitive psychology","level":1,"score":0.27239999175071716},{"id":"https://openalex.org/C158251709","wikidata":"https://www.wikidata.org/wiki/Q354025","display_name":"Intrusion","level":2,"score":0.2687000036239624},{"id":"https://openalex.org/C41065033","wikidata":"https://www.wikidata.org/wiki/Q2825412","display_name":"Adversary","level":2,"score":0.25940001010894775},{"id":"https://openalex.org/C3017944768","wikidata":"https://www.wikidata.org/wiki/Q1450463","display_name":"Poison control","level":2,"score":0.2556999921798706},{"id":"https://openalex.org/C2777508537","wikidata":"https://www.wikidata.org/wiki/Q7936620","display_name":"Visual reasoning","level":2,"score":0.2542000114917755},{"id":"https://openalex.org/C140547941","wikidata":"https://www.wikidata.org/wiki/Q7797194","display_name":"Threat model","level":2,"score":0.2533000111579895}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2603.03205","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.03205","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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.2603.03205","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.03205","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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":[{"display_name":"Peace, Justice and strong institutions","id":"https://metadata.un.org/sdg/16","score":0.8263720273971558}],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Agentic":[0],"language":[1],"models":[2],"operate":[3],"in":[4,50],"a":[5,23,67,89],"fundamentally":[6],"different":[7],"safety":[8,80,99,122],"regime":[9],"than":[10],"chat":[11],"models:":[12],"they":[13],"must":[14],"plan,":[15,90],"call":[16],"tools,":[17],"and":[18,45,60,83,101,139,141,153,177,190],"execute":[19],"long-horizon":[20],"actions":[21],"where":[22],"single":[24],"misstep,":[25],"such":[26],"as":[27,88,103],"accessing":[28],"files":[29],"or":[30,94,179],"entering":[31],"credentials,":[32],"can":[33],"cause":[34],"irreversible":[35],"harm.":[36],"Existing":[37],"alignment":[38],"methods,":[39],"largely":[40],"optimized":[41],"for":[42,73],"static":[43],"generation":[44],"task":[46,182],"completion,":[47],"break":[48],"down":[49],"these":[51],"settings":[52],"due":[53],"to":[54,163],"sequential":[55],"decision-making,":[56],"adversarial":[57],"tool":[58,76,151],"feedback,":[59],"overconfident":[61],"intermediate":[62],"reasoning.":[63],"We":[64,129],"introduce":[65],"MOSAIC,":[66],"post-training":[68],"framework":[69],"that":[70],"aligns":[71],"agents":[72],"safe":[74],"multi-step":[75],"use":[77,112],"by":[78,126,161,168],"making":[79],"decisions":[81],"explicit":[82,98],"learnable.":[84],"MOSAIC":[85,131,157],"structures":[86],"inference":[87],"check,":[91],"then":[92],"act":[93],"refuse":[95],"loop,":[96],"with":[97,116],"reasoning":[100],"refusal":[102,167],"first-class":[104],"actions.":[105],"To":[106],"train":[107],"without":[108],"trajectory-level":[109],"labels,":[110],"we":[111],"preference-based":[113],"reinforcement":[114],"learning":[115],"pairwise":[117],"trajectory":[118],"comparisons,":[119],"which":[120],"captures":[121],"distinctions":[123],"often":[124],"missed":[125],"scalar":[127],"rewards.":[128],"evaluate":[130],"zero-shot":[132],"across":[133,142,187],"three":[134],"model":[135],"families,":[136],"Qwen2.5-7B,":[137],"Qwen3-4B-Thinking,":[138],"Phi-4,":[140],"out-of-distribution":[143],"benchmarks":[144],"spanning":[145],"harmful":[146,159],"tasks,":[147],"prompt":[148],"injection,":[149],"benign":[150,181],"use,":[152],"cross-domain":[154],"privacy":[155,175],"leakage.":[156],"reduces":[158],"behavior":[160],"up":[162],"50%,":[164],"increases":[165],"harmful-task":[166],"over":[169],"20%":[170],"on":[171],"injection":[172],"attacks,":[173],"cuts":[174],"leakage,":[176],"preserves":[178],"improves":[180],"performance,":[183],"demonstrating":[184],"robust":[185],"generalization":[186],"models,":[188],"domains,":[189],"agentic":[191],"settings.":[192]},"counts_by_year":[],"updated_date":"2026-07-28T07:46:37.118299","created_date":"2026-03-05T00:00:00"}