Energy-Based-CrossAttention

用能量模型优化注意力,让文生图更懂上下文

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 图像生成 🍴 复刻 6 🐛 待处理 3 👀 关注 2
GitHub ↗
GitHub 星标 ★ 51
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,可自行部署使用,无收费计划。

项目预览

Energy-Based-CrossAttention 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类图像生成
开发团队EnergyAttention
定价模式🏷️完全免费
价格说明开源项目,可自行部署使用,无收费计划。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
主要语言Python
技术栈/模型cross-attention,ebm,energy-based-model,pytorch,stable-diffusion
GitHub 星标★ 51
上线时间2023-05-24 00:00:00
代码更新2024-04-01
仓库体积15.7 MB
本站数据更新2026-11-08
维护状态低维护
收录时间2026-08-09
浏览次数4

项目介绍

Energy-Based Cross Attention(EBCA)是一个用于文本到图像扩散模型的官方实现,旨在通过能量基交叉注意力机制实现贝叶斯上下文更新。该项目解决的是扩散模型在生成图像时,文本条件与图像特征对齐不够精确、上下文信息利用不充分的问题。其核心能力包括:将交叉注意力层重新建模为能量基模型,通过能量函数动态调整注意力权重,从而在生成过程中更有效地融合文本语义与视觉特征;支持与Stable Diffusion等主流扩散模型集成,提供训练和推理代码。项目基于PyTorch实现,代码结构清晰,适合研究扩散模型注意力机制的开发者使用。目前处于早期阶段,星标数较少,但提供了完整的官方实现,便于复现论文结果。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 51

技术标签

核心亮点

  • 提出能量基交叉注意力新机制,理论上能提升文本-图像对齐精度
  • 官方实现,代码与论文同步,便于复现研究结果
  • 基于PyTorch,易于集成到现有Stable Diffusion工作流

不足之处

  • 项目处于早期,文档和示例较少,上手门槛较高
  • 依赖特定扩散模型版本,兼容性需自行验证

适用场景

  • 学术研究:探索扩散模型注意力机制改进
  • 文生图应用:提升复杂文本提示的生成准确性
  • 模型微调:在特定数据集上训练定制化生成模型

替代项目

CrossAttention (Stable Diffusion内置)、FreeU、AttnGAN

同类项目推荐

pixelCNN 开源

用 Theano 逐像素生成图像,理解自回归模型原理

Theano reimplementation of pixelCNN architecture

★ 166 2026-08-09
contrastive-unpaired-translation 开源

不用配对数据,训练速度比CycleGAN快一倍,图像翻译效果更好

Contrastive unpaired image-to-image translation, faster and lighter training than cy···

★ 2564 2026-08-09
GALIP 开源

更快更小的文生图模型,大规模训练不妥协

[CVPR2023] A faster, smaller, and better text-to-image model for large-scale trainin···

★ 251 2026-08-09
ReNoise-Inversion 开源

让真实图像精准反演,编辑更自然可控

Officail Implementation for "ReNoise: Real Image Inversion Through Iterative Noising···

★ 263 2026-08-09
nix-stable-diffusion 开源

在 NixOS 上用 Flake 一键跑起 Stable Diffusion,环境不再折腾。

Flake for running SD on NixOS

★ 119 2026-08-09
Fooocus-zh 开源

给 Fooocus 装上中文皮肤,AI 绘画更顺手

Fooocus 汉化补丁 #Fooocus

★ 113 2026-08-09
RvR 开源

让图像编辑变成再生成,效果更自然更精准。

Regeneration over editing: unlocking more effective image refinement!

★ 54 2026-08-09
gpt_image_playground 开源

用 GPT 图像模型,快速生成和编辑图片,开箱即用

支持 GPT Image、Gemini(Nano Banana)和 Grok 的图片生成与编辑工具

★ 4045 2026-08-09