展开菜单
首页 精品内容 本月促销 装机必备 Windows PDF教程 macOS软件 IOS软件 Android AI 专题
全部分类

当前位置:

首页 > 硬件相关 > 社区项目 ROSA-Tuning:验证 RWKV-8 ROSA 效果

社区项目 ROSA-Tuning:验证 RWKV-8 ROSA 效果

ROSA-Tuning为LLM引入CPU检索模块作为并行于注意力的外挂记忆体,可学习注入隐状态,兼顾长上下文表达力与效率。PG-19上窗口注意力结合ROSA困惑度17.63,优于全局注意力;LongBench恢复96.5%性能,大海捞针召回100%。训练时冻结原始参数。

本文目录
    # ROSA-Tuning:给大模型装上一个“外设”记忆体,长上下文难题的新解法 今天要聊的这个项目,来自 RWKV 社区开发者 zyaaa-ux,项目地址已经放在下面了。需要先说明的是,这是社区自主实现的 ROSA 方案,并不代表 RWKV-8 ROSA 的真实性能表现,相关效果仅供参考。 先说几个核心判断:长上下文建模一直是预训练模型落地时的硬骨头。窗口注意力虽然省资源,但长程依赖一上来就垮;全局注意力效果虽好,计算开销又让人望而却步。有没有第三条路?ROSA-Tuning 给出的答案是:加一个基于 CPU 的检索模块,在标准注意力结构之外并行工作,从超长上下文中快速定位历史位置,再把检索结果以可学习的方式注入模型隐状态。后续的状态融合则交给轻量的受限注意力来完成——兼顾表达力与效率。 这套方案的架构图后面会贴出来,先看性能数据。 ## 性能测试 ### 困惑度(PPL)对比 PG-19 数据集上的结果很有意思。窗口注意力直接把 PPL 推到了 74.50,而加上 ROSA 适配器后,不仅逆转了这个趋势,最终指标甚至优于全局注意力基线。 | Model | PPL (越低越好) | |-------|----------------| | Global Attention | 18.96 | | Windowed Attention | 74.50 | | **Windowed + ROSA** | **17.63** | 实验配置方面,基础模型用的是 Qwen3-Base-0.6B,包含全局注意力和窗口注意力两个版本。训练在 PG-19 训练集上单轮完成,共 28,000 个样本,原始模型参数全部冻结,只优化 ROSA 适配器。测试时输入序列长度 16k,窗口大小 1024。 ### 长文本理解能力(LongBench) 在“大海捞针”(NIAH)任务上,ROSA 拿下了满分 100% 的精准召回。LongBench 综合得分恢复到了全局注意力性能的 96.5%——注意,这是在计算开销几乎不变的前提下做到的。 | Task / Metric | Global Attention | Windowed (2048) | **Windowed + ROSA** | |---------------|-----------------|-----------------|---------------------| | **NIAH (大海捞针)** | **100.00** | 6.20 | **100.00** | | **TriviaQA** | 86.20 | 61.56 | **84.34** | | **Multi_news** | 23.23 | 10.43 | **23.76** | | **Samsum** | 42.04 | 32.51 | **40.53** | | **TREC** | 72.67 | 52.67 | 68.00 | | **Gov_report** | 31.11 | 13.08 | 26.19 | | **LongBench 平均分** | **59.21** | 29.41 | **57.14** | 这次实验的基础模型换成了 Qwen3-1.7B-Base,窗口尺寸 2048。训练数据总计约 37B tokens,其中约 30B 来自 prolong 数据集,其余来自其他长上下文推理任务数据集,并且严格避免了与测试集重叠。 ## 使用方法 项目作者已经完成了大量验证实验,这里重点介绍截至 2025 年 12 月 29 日发布的 `2025.12.29 qkv_update.py` 版本的部署流程。 需要注意:请提前将 Hugging Face datasets 库加载的数据以 Arrow 格式本地化存储。 ### 环境准备与代码拉取 首先安装必要依赖: ```bash pip install torch transformers datasets deepspeed numba numpy ``` 可选安装 `flash-attn` 加速库(首次安装需编译),能进一步提升运行速度。 然后获取源码: ```bash git clone https://www.php.cn/link/c1b952b6948f085d619846108cec1b8b ``` ### DeepSpeed 配置文件准备 项目采用 DeepSpeed 进行分布式训练加速,需要在本地创建 `deepspeed_config.json` 配置文件,参考内容如下: ```json { "fp16": { "enabled": "auto", "loss_scale": 0, "loss_scale_window": 1000, "initial_scale_power": 16, "hysteresis": 2, "min_loss_scale": 1 }, "bf16": { "enabled": "auto" }, "zero_optimization": { "stage": 2, "allgather_partitions": true, "allgather_bucket_size": 200000000, "overlap_comm": true, "reduce_scatter": true, "reduce_bucket_size": 200000000, "contiguous_gradients": true, "offload_optimizer": { "device": "cpu", "pin_memory": true }, "offload_param": { "device": "none" } }, "gradient_accumulation_steps": "auto", "train_batch_size": "auto", "train_micro_batch_size_per_gpu": "auto", "gradient_clipping": "auto", "steps_per_print": 20, "wall_clock_breakdown": false } ``` 如果 GPU 显存充足,建议移除 `offload_optimizer` 中的 `pin_memory` 字段,并将 `device` 改为 `"none"`,训练速度会更快。 ### 参数配置修改 `2025.12.29 qkv_update.py` 文件第 68~73 行定义了关键路径变量,需要按实际环境替换为本地路径: ```python MODEL_LOCAL_DIR = "/path/to/base/model/" # 本地基础模型路径 MODEL_DIR = "/path/to/checkpoint/" # 模型检查点保存路径 DATASET_DIR = "/path/to/processed/dataset/" # 数据集路径 OUTPUT_DIR = "/path/to/output/" # 输出路径 DEEPSPEED_CONFIG_PATH = "/path/to/deepspeed/config.json" # DeepSpeed 配置文件路径 ``` 如果需要进一步降低显存占用,可以将第 119 行设为 `True`,启用梯度累积: ```python GRADIENT_CHECKPOINTING = True # 默认为 False ``` 如果没有安装 `flash-attn`,请将第 78 行设为 `False`,禁用该加速特性: ```python USE_FLASH_ATTN = False # 默认为 True ``` ### 启动训练命令 项目内嵌了 DeepSpeed 分布式逻辑(如 `is_main_process` 等判断),推荐统一使用 `deepspeed` 命令启动: ```bash deepspeed --num_gpus=1 2025.12.29 qkv_update.py ``` 成功启动后,终端会输出如下日志界面: 上图是在单张 RTX 4090 上使用 200 条长度为 128 的样本进行流程验证的截图;实际训练 16k 长度文本时对显存要求较高。 ## 原理概述 ## 附:RWKV 社区相关资源 RWKV 社区持续围绕模型进行开源共建,感兴趣的读者可以通过以下渠道了解更多信息: - RWKV 中文文档:https://www.php.cn/link/ad627bf5fd6966693e97a7349d85589c - RWKV 论坛:https://www.php.cn/link/ca66c4195dbebc6f59ceaf0e10629664 源码地址:点击下载
    本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系bd@zhengruan.com
    作者最新文章
    硬件相关
    相关文章 更多
    精品专题 更多
    装机必备

    正软商城装机必备专区,精选办公、浏览器、安全防护、影音播放、压缩解压、设计创作和系统工具等电脑常用正版软件,帮助用户快速完成新电脑软件配置。

    Windows

    正软商城Windows软件专区,汇集适用于Windows电脑的办公、设计、安全防护、影音播放、开发工具和系统优化软件,提供软件介绍、系统要求、正版授权及购买下载服务。

    PDF教程

    正软商城PDF教程频道提供PDF编辑、转换、合并、拆分、压缩及格式处理方法,同时介绍常用PDF软件和工具的使用技巧。

    macOS软件

    正软商城macOS软件专区,精选适用于Mac电脑的办公、设计、影音、效率、开发和系统工具,提供软件功能介绍、macOS兼容版本、正版授权及购买下载服务。

    IOS软件

    正软商城iOS软件专区,精选适用于iPhone和iPad的办公、学习、影音、设计、效率及AI应用,提供功能介绍、适用设备、系统要求和正版获取方式等信息。

    AI

    正软商城AI软件专区,汇集AI写作、AI绘画、AI视频、AI办公、AI编程、AI翻译、智能客服和数据分析等人工智能工具,提供功能介绍、适用平台、收费方式及正版购买信息。

    Mac软件 更多
    Postman macOS版
    Postman macOS版

    Postman是一套面向API全生命周期的开发与测试平台,可在Mac上发送和调试多种协议请求,管理集合、环境及认证信息,并完成脚本测试、模拟服务、监控、文档发布和团队协作。它适合开发、测试及运维人员使用,提供个人免费方案。

    Docker Desktop macOS版
    Docker Desktop macOS版

    Docker Desktop 是 Docker 官方为 Mac 提供的容器开发环境,集成 Docker Engine、命令行工具、Compose、Build、Kubernetes 与图形化管理界面。

    Android Studio macOS版
    Android Studio macOS版

    Android Studio 是 Google 面向安卓应用开发提供的官方集成开发环境,基于 IntelliJ IDEA 构建。它整合智能代码编辑、Gradle 构建、界面预览、设备模拟、性能分析、调试测试及 APK 分析等工具。

    Xcode macOS版
    Xcode macOS版

    Xcode 是苹果面向 Mac 提供的官方集成开发环境,将代码编辑、项目构建、SwiftUI 实时预览、设备模拟、自动化测试、断点调试、性能分析和应用分发整合在同一套工作流中,适合开发与维护 iPhone、iPad、Mac、Apple Wa

    JetBrains WebStorm macOS版
    JetBrains WebStorm macOS版

    JetBrains WebStorm 是面向 JavaScript、TypeScript 和现代 Web 技术的专业集成开发环境,可智能理解项目结构,提供代码补全、重构、错误检查、运行调试、测试及 Git 工具。

    PhpStorm macOS版
    PhpStorm macOS版

    PhpStorm 是 JetBrains 推出的专业 PHP 集成开发环境,可在 Mac 上完成代码编写、智能检查、重构、调试、测试和数据库管理。它支持主流 PHP 框架、Composer、Git、Docker 与远程解释器。

    JetBrains PyCharm macOS版
    JetBrains PyCharm macOS版

    JetBrains PyCharm 是面向 Python 开发者的集成开发环境,提供智能补全、代码检查、重构、调试、测试、版本控制及虚拟环境管理功能。Mac 版兼容 Intel 与 Apple 芯片,核心功能可免费使用。

    JetBrains IntelliJ IDEA macOS版
    JetBrains IntelliJ IDEA macOS版

    IntelliJ IDEA 是 JetBrains 推出的专业集成开发环境,重点服务于 Java、Kotlin 与 JVM 项目开发。软件集成代码补全、静态检查、重构、调试、测试、版本控制及构建工具,并可借助插件扩展语言和框架支持。

    Visual Studio Code macOS版
    Visual Studio Code macOS版

    Visual Studio Code 是微软提供的免费代码编辑器,支持 macOS,并为 Intel 与 Apple 芯片 Mac 提供适配版本。软件集代码编辑、调试、Git 源代码管理、集成终端和扩展市场于一体。

    REAPER macOS版
    REAPER macOS版

    REAPER是Cockos开发的数字音频工作站,提供多轨音频与MIDI录制、剪辑、处理、混音和母带制作工具。Mac版兼容Intel与Apple芯片,支持AU、VST、VST3、CLAP等插件格式,并提供高度可定制的工作流程。

    Adobe Audition macOS版
    Adobe Audition macOS版

    Adobe Audition 是面向播客、影视后期、音乐制作及内容创作者的专业音频工作站,提供波形精修、多轨录制与混音、频谱编辑、降噪修复、响度匹配和效果处理工具,并可与 Adobe Premiere 协同完成视频声音制作。

    Pro Tools macOS版
    Pro Tools macOS版

    Pro Tools 是 Avid 推出的专业数字音频工作站,覆盖多轨录音、无损剪辑、MIDI 编曲、插件处理、自动化混音及声音与画面同步等工作。它拥有 Intro、Artist、Studio 和 Ultimate 等方案。

    WINDOWS 更多
    PixPix 网页版
    PixPix 网页版

    PixPix 是面向品牌、卖家和内容团队的在线 AI 图片视频创作平台,提供图视频 Agent、无限画布、图片生成、视频生成、自然语言编辑及商品套图、产品精修和带货视频等电商工具。

    Postman Windows版
    Postman Windows版

    Postman 是面向 API 全生命周期的开发与协作平台。Windows 桌面版可用于构建和发送 HTTP、GraphQL、gRPC、WebSocket、MQTT 等请求,并提供环境变量、集合、脚本测试、模拟服务、监控、文档及团队工作区等

    Docker Desktop Windows版
    Docker Desktop Windows版

    Docker Desktop 为 Windows 开发者提供集成式容器环境,内置 Docker Engine、命令行工具、Compose、Build 与 Kubernetes 等组件,可在本机完成镜像构建、容器运行、日志查看和资源管理。

    Android Studio Windows版
    Android Studio Windows版

    Android Studio 是 Google 面向 Android 应用开发提供的官方集成开发环境,以 IntelliJ IDEA 为基础,集成智能代码编辑、Gradle 构建、设备模拟、界面预览、调试测试及性能分析工具,

    JetBrains WebStorm Windows版
    JetBrains WebStorm Windows版

    WebStorm 是 JetBrains 面向 JavaScript、TypeScript 和现代 Web 技术打造的集成开发环境。它提供项目级代码分析、智能补全、安全重构、导航搜索,并集成调试、测试、终端、Git 与常用前端工具,

    JetBrains PhpStorm Windows版
    JetBrains PhpStorm Windows版

    PhpStorm 是 JetBrains 面向 PHP 与现代 Web 开发打造的专业集成开发环境,提供代码补全、静态检查、重构、调试、测试、数据库、版本控制及前端语言支持。Windows 用户可直接下载安装程序,

    JetBrains PyCharm Windows版
    JetBrains PyCharm Windows版

    PyCharm 是 JetBrains 推出的 Python 集成开发环境,提供智能代码补全、错误检查、重构、调试、测试、Git、终端及 Jupyter Notebook 支持。统一版安装后可体验 30 天 Pro 功能,

    JetBrains IntelliJ IDEA Windows版
    JetBrains IntelliJ IDEA Windows版

    JetBrains IntelliJ IDEA 是以 Java、Kotlin 和 JVM 项目为核心的专业集成开发环境,提供智能补全、代码检查、安全重构、调试、测试、版本控制及构建工具集成。统一发行版的核心功能可免费用于商业和非商业项目,

    Visual Studio Code Windows版
    Visual Studio Code Windows版

    Visual Studio Code 是微软推出的免费代码编辑器,Windows 版集成智能补全、代码导航、调试、终端和版本控制,并可通过扩展添加语言、格式化及远程开发能力。它启动较快、配置灵活,

    REAPER Windows版
    REAPER Windows版

    REAPER 是 Cockos 推出的数字音频工作站,适用于录音、MIDI 编曲、音频编辑、混音、母带制作及声音设计。软件安装包小巧,支持多轨录制、灵活路由、自动化、第三方插件和脚本扩展,并提供完整功能的 60 天免费评估期。

    Adobe Audition Windows版
    Adobe Audition Windows版

    Adobe Audition 是面向播客、影视后期、广播与音乐制作的专业音频工作站,提供波形编辑、多轨混音、频谱分析、录音、降噪修复和效果处理等功能。Windows 版通过 Adobe Creative Cloud 安装和管理,

    Pro Tools Windows版
    Pro Tools Windows版

    Pro Tools 是 Avid 推出的专业数字音频工作站,提供多轨录音、音频剪辑、MIDI编曲、插件处理、自动化混音及影视声音制作能力。它适合音乐人、录音师、制作人和音频后期团队使用,并提供免费入门版、订阅方案及永久授权选择。