{"id":"https://openalex.org/W7148672232","doi":"https://doi.org/10.48550/arxiv.2604.01826","title":"SafeRoPE: Risk-specific Head-wise Embedding Rotation for Safe Generation in Rectified Flow Transformers","display_name":"SafeRoPE: Risk-specific Head-wise Embedding Rotation for Safe Generation in Rectified Flow Transformers","publication_year":2026,"publication_date":"2026-04-02","ids":{"openalex":"https://openalex.org/W7148672232","doi":"https://doi.org/10.48550/arxiv.2604.01826"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2604.01826","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.01826","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.01826","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5101766153","display_name":"Xiang Yang","orcid":"https://orcid.org/0009-0002-0727-7638"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yang, Xiang","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5132841641","display_name":"Feifei Li","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Li, Feifei","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5132851773","display_name":"Mi Zhang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Mi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5129691329","display_name":"Geng Hong","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Hong, Geng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5132871885","display_name":"Xiaoyu You","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"You, Xiaoyu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5132908949","display_name":"Min Yang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yang, Min","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/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.7577999830245972,"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"}},"topics":[{"id":"https://openalex.org/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.7577999830245972,"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/T11714","display_name":"Multimodal Machine Learning Applications","score":0.0471000000834465,"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/T11307","display_name":"Domain Adaptation and Few-Shot Learning","score":0.031199999153614044,"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/embedding","display_name":"Embedding","score":0.5720000267028809},{"id":"https://openalex.org/keywords/transformer","display_name":"Transformer","score":0.51419997215271},{"id":"https://openalex.org/keywords/linear-subspace","display_name":"Linear subspace","score":0.4632999897003174},{"id":"https://openalex.org/keywords/semantics","display_name":"Semantics (computer science)","score":0.4564000070095062},{"id":"https://openalex.org/keywords/overhead","display_name":"Overhead (engineering)","score":0.4284999966621399},{"id":"https://openalex.org/keywords/fidelity","display_name":"Fidelity","score":0.3878999948501587},{"id":"https://openalex.org/keywords/projection","display_name":"Projection (relational algebra)","score":0.36800000071525574}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7129999995231628},{"id":"https://openalex.org/C41608201","wikidata":"https://www.wikidata.org/wiki/Q980509","display_name":"Embedding","level":2,"score":0.5720000267028809},{"id":"https://openalex.org/C66322947","wikidata":"https://www.wikidata.org/wiki/Q11658","display_name":"Transformer","level":3,"score":0.51419997215271},{"id":"https://openalex.org/C12362212","wikidata":"https://www.wikidata.org/wiki/Q728435","display_name":"Linear subspace","level":2,"score":0.4632999897003174},{"id":"https://openalex.org/C184337299","wikidata":"https://www.wikidata.org/wiki/Q1437428","display_name":"Semantics (computer science)","level":2,"score":0.4564000070095062},{"id":"https://openalex.org/C2779960059","wikidata":"https://www.wikidata.org/wiki/Q7113681","display_name":"Overhead (engineering)","level":2,"score":0.4284999966621399},{"id":"https://openalex.org/C2776459999","wikidata":"https://www.wikidata.org/wiki/Q2119376","display_name":"Fidelity","level":2,"score":0.3878999948501587},{"id":"https://openalex.org/C57493831","wikidata":"https://www.wikidata.org/wiki/Q3134666","display_name":"Projection (relational algebra)","level":2,"score":0.36800000071525574},{"id":"https://openalex.org/C177264268","wikidata":"https://www.wikidata.org/wiki/Q1514741","display_name":"Set (abstract data type)","level":2,"score":0.3379000127315521},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.3294999897480011},{"id":"https://openalex.org/C26517878","wikidata":"https://www.wikidata.org/wiki/Q228039","display_name":"Key (lock)","level":2,"score":0.321399986743927},{"id":"https://openalex.org/C150807984","wikidata":"https://www.wikidata.org/wiki/Q1992074","display_name":"Bit array","level":3,"score":0.320499986410141},{"id":"https://openalex.org/C124101348","wikidata":"https://www.wikidata.org/wiki/Q172491","display_name":"Data mining","level":1,"score":0.31450000405311584},{"id":"https://openalex.org/C74050887","wikidata":"https://www.wikidata.org/wiki/Q848368","display_name":"Rotation (mathematics)","level":2,"score":0.30640000104904175},{"id":"https://openalex.org/C83665646","wikidata":"https://www.wikidata.org/wiki/Q42139305","display_name":"Feature vector","level":2,"score":0.3021000027656555},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.3000999987125397},{"id":"https://openalex.org/C167966045","wikidata":"https://www.wikidata.org/wiki/Q5532625","display_name":"Generative model","level":3,"score":0.28540000319480896},{"id":"https://openalex.org/C1893757","wikidata":"https://www.wikidata.org/wiki/Q3653001","display_name":"Inversion (geology)","level":3,"score":0.2842999994754791},{"id":"https://openalex.org/C149364088","wikidata":"https://www.wikidata.org/wiki/Q185917","display_name":"Translation (biology)","level":4,"score":0.28040000796318054},{"id":"https://openalex.org/C113775141","wikidata":"https://www.wikidata.org/wiki/Q428691","display_name":"Computer engineering","level":1,"score":0.27570000290870667},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.2728999853134155}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2604.01826","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.01826","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.01826","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.01826","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":{"Recent":[0],"Text-to-Image":[1],"(T2I)":[2],"models":[3,57],"based":[4],"on":[5,32,204],"rectified-flow":[6],"transformers":[7],"(e.g.,":[8,58],"SD3,":[9],"FLUX)":[10],"achieve":[11],"high":[12],"generative":[13],"fidelity":[14],"but":[15],"remain":[16],"vulnerable":[17],"to":[18,48,54,111,199],"unsafe":[19,77,97,148,152,182,214],"semantics,":[20],"especially":[21],"when":[22],"triggered":[23],"by":[24,128,150],"multi-token":[25],"interactions.":[26],"Existing":[27],"mitigation":[28,233],"methods":[29],"largely":[30],"rely":[31],"fine-tuning":[33],"or":[34,188],"attention":[35,70],"modulation":[36],"for":[37,96,141,164,237],"concept":[38],"unlearning;":[39],"however,":[40],"their":[41],"expensive":[42],"computational":[43],"overhead":[44],"and":[45,74,114,136,157,196,206,234],"design":[46],"tailored":[47],"U-Net-based":[49],"denoisers":[50],"hinder":[51],"direct":[52],"adaptation":[53],"transformer-based":[55],"diffusion":[56],"MMDiT).":[59],"In":[60],"this":[61],"paper,":[62],"we":[63,131],"conduct":[64],"an":[65],"in-depth":[66],"analysis":[67],"of":[68,91,213,240],"the":[69,105,112,124],"mechanism":[71],"in":[72,123,228],"MMDiT":[73],"find":[75],"that":[76,103,179,223],"semantics":[78,183],"concentrate":[79],"within":[80,154],"interpretable,":[81],"low-dimensional":[82],"subspaces":[83,149],"at":[84,245],"head":[85],"level,":[86],"where":[87],"a":[88,134,159],"finite":[89],"set":[90],"safety-critical":[92,155],"heads":[93],"is":[94],"responsible":[95],"feature":[98],"extraction.":[99],"We":[100,173],"further":[101],"observe":[102],"perturbing":[104],"Rotary":[106],"Positional":[107],"Embedding":[108],"(RoPE)":[109],"applied":[110],"query":[113,205],"key":[115,207],"vectors":[116],"can":[117,180],"effectively":[118],"modify":[119],"some":[120],"specific":[121],"concepts":[122],"generated":[125],"images.":[126],"Motivated":[127],"these":[129,171],"insights,":[130],"propose":[132],"SafeRoPE,":[133],"lightweight":[135],"fine-grained":[137],"safe":[138,238],"generation":[139,218,239],"framework":[140],"MMDiT.":[142,241],"Specifically,":[143],"SafeRoPE":[144,191,224],"first":[145],"constructs":[146],"head-wise":[147,176,194,202],"decomposing":[151],"embeddings":[153],"heads,":[156],"computes":[158],"Latent":[160],"Risk":[161],"Score":[162],"(LRS)":[163],"each":[165],"input":[166],"vector":[167,208],"via":[168],"projection":[169],"onto":[170],"subspaces.":[172],"then":[174],"introduce":[175],"RoPE":[177,197],"perturbations":[178,198],"suppress":[181],"without":[184],"degrading":[185],"benign":[186],"content":[187,232],"image":[189],"quality.":[190],"combines":[192],"both":[193],"LRS":[195],"perform":[200],"risk-specific":[201],"rotation":[203],"embeddings,":[209],"enabling":[210],"precise":[211],"suppression":[212],"outputs":[215],"while":[216],"maintaining":[217],"fidelity.":[219],"Extensive":[220],"experiments":[221],"demonstrate":[222],"achieves":[225],"SOTA":[226],"performance":[227],"balancing":[229],"effective":[230],"harmful":[231],"utility":[235],"preservation":[236],"Codes":[242],"are":[243],"available":[244],"https://github.com/deng12yx/SafeRoPE.":[246]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-04-04T00:00:00"}