AttenST: A Training-Free Attention-Driven Style Transfer Framework with Pre-Trained Diffusion Models

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Huang, Bo, Xu, Wenlun, Han, Qizhuo, Jing, Haodong, Li, Ying
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866917949999153152
author Huang, Bo
Xu, Wenlun
Han, Qizhuo
Jing, Haodong
Li, Ying
author_facet Huang, Bo
Xu, Wenlun
Han, Qizhuo
Jing, Haodong
Li, Ying
contents While diffusion models have achieved remarkable progress in style transfer tasks, existing methods typically rely on fine-tuning or optimizing pre-trained models during inference, leading to high computational costs and challenges in balancing content preservation with style integration. To address these limitations, we introduce AttenST, a training-free attention-driven style transfer framework. Specifically, we propose a style-guided self-attention mechanism that conditions self-attention on the reference style by retaining the query of the content image while substituting its key and value with those from the style image, enabling effective style feature integration. To mitigate style information loss during inversion, we introduce a style-preserving inversion strategy that refines inversion accuracy through multiple resampling steps. Additionally, we propose a content-aware adaptive instance normalization, which integrates content statistics into the normalization process to optimize style fusion while mitigating the content degradation. Furthermore, we introduce a dual-feature cross-attention mechanism to fuse content and style features, ensuring a harmonious synthesis of structural fidelity and stylistic expression. Extensive experiments demonstrate that AttenST outperforms existing methods, achieving state-of-the-art performance in style transfer dataset.
format Preprint
id arxiv_https___arxiv_org_abs_2503_07307
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle AttenST: A Training-Free Attention-Driven Style Transfer Framework with Pre-Trained Diffusion Models
Huang, Bo
Xu, Wenlun
Han, Qizhuo
Jing, Haodong
Li, Ying
Computer Vision and Pattern Recognition
While diffusion models have achieved remarkable progress in style transfer tasks, existing methods typically rely on fine-tuning or optimizing pre-trained models during inference, leading to high computational costs and challenges in balancing content preservation with style integration. To address these limitations, we introduce AttenST, a training-free attention-driven style transfer framework. Specifically, we propose a style-guided self-attention mechanism that conditions self-attention on the reference style by retaining the query of the content image while substituting its key and value with those from the style image, enabling effective style feature integration. To mitigate style information loss during inversion, we introduce a style-preserving inversion strategy that refines inversion accuracy through multiple resampling steps. Additionally, we propose a content-aware adaptive instance normalization, which integrates content statistics into the normalization process to optimize style fusion while mitigating the content degradation. Furthermore, we introduce a dual-feature cross-attention mechanism to fuse content and style features, ensuring a harmonious synthesis of structural fidelity and stylistic expression. Extensive experiments demonstrate that AttenST outperforms existing methods, achieving state-of-the-art performance in style transfer dataset.
title AttenST: A Training-Free Attention-Driven Style Transfer Framework with Pre-Trained Diffusion Models
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2503.07307