用脑信号指导智能体训练,让神经反馈替代人工打分。
Towards Reinforcement Learning from Neural Feedback: Mapping fNIRS Signals to Agent Performance
- 通过fNIRS信号分类判断智能体表现优劣,实现隐式反馈。
- 多任务下二分类F1达67%,多分类46%,跨人泛化可提升41%。
- 适合做脑机接口、无监督强化学习的科研与工程人员。
强化学习中的人类反馈(RLHF)通过整合用户反馈来对齐智能体行为与人类偏好。本文提出一种通过隐式神经信号引导智能体训练的框架,聚焦于神经信号分类问题。我们构建并公开了一个新数据集,包含25名参与者在三个任务场景(抓取放置机器人、月球着陆器、Flappy Bird)中的功能近红外光谱(fNIRS)记录。通过训练多个分类器,基于预处理后的fNIRS特征窗口预测不同水平的智能体表现(最优、次优或最差),在跨条件与跨域测试中,二分类平均F1为67%,多分类为46%。同时训练多个回归器,以连续方式预测智能体动作与近最优策略之间的偏差程度。最后评估了跨被试泛化能力,结果显示,仅用少量个体数据微调预训练模型,可使二分类和多分类的平均F1分别提升17%和41%。结果表明,将隐式fNIRS信号映射到智能体性能是可行且可优化的,为未来神经反馈强化学习(RLNF)系统奠定基础。
原文摘要 · Abstract (English)
Reinforcement Learning from Human Feedback (RLHF) is a methodology that aligns agent behavior with human preferences by integrating user feedback into the agent's training process. This paper introduces a framework that guides agent training through implicit neural signals, with a focus on the neural classification problem. Our work presents and releases a novel dataset of functional near-infrared spectroscopy (fNIRS) recordings collected from 25 human participants across three domains: Pick-and-Place Robot, Lunar Lander, and Flappy Bird. We train multiple classifiers to predict varying levels of agent performance (optimal, suboptimal, or worst-case) from windows of preprocessed fNIRS features, achieving an average F1 score of 67% for binary and 46% for multi-class classification across conditions and domains. We also train multiple regressors to predict the degree of deviation between an agent's chosen action and a set of near-optimal policy actions, providing a continuous measure of performance. Finally, we evaluate cross-subject generalization and show that fine-tuning pre-trained models with a small sample of subject-specific data increases average F1 scores by 17% and 41% for binary and multi-class models, respectively. Our results demonstrate that mapping implicit fNIRS signals to agent performance is feasible and can be improved, laying the foundation for future Reinforcement Learning from Neural Feedback (RLNF) systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。