1. 多目标强化学习算法复现概述
最近在复现Runzhe Yang等人提出的通用多目标强化学习算法时,遇到了不少环境配置和代码适配的问题。这篇论文提出的envelope MOQ-learning算法在多个基准测试中表现出色,包括两个合成环境(深海寻宝DST和果树导航FTN)和两个真实环境(对话策略学习和超级马里奥游戏)。作为一位长期从事强化学习研究的工程师,我想分享这次在RTX 4060显卡上复现FTN环境的完整过程,特别是针对现代硬件环境所做的必要调整。
多目标强化学习(MORL)与传统的单目标RL不同,它需要同时优化多个可能相互冲突的目标函数。在FTN环境中,智能体需要在树形导航中收集6种不同水果,每种水果代表一个独立的目标。envelope方法的核心创新在于它能学习一个参数化的策略集合,可以覆盖整个Pareto前沿,而不需要为每个权重组合单独训练策略。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 环境配置与依赖安装
2.1 PyTorch版本适配问题
原代码基于PyTorch 0.4.0开发,这对于RTX 40系列显卡存在严重的兼容性问题。RTX 4060采用Ada Lovelace架构(sm_89),而PyTorch 0.4.0最高仅支持到Turing架构(sm_75)。经过多次尝试,我确定了以下可行的配置方案:
bash复制# 创建专用Python环境
conda create -n morl-synthetic python=3.6.15 -y
conda activate morl-synthetic
# 安装基础依赖(注意版本兼容性)
pip install numpy==1.19.5 scipy==1.5.4 -i https://pypi.tuna.tsinghua.edu.cn/simple
# 安装适配RTX4060的PyTorch版本
pip install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio==0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
注意:Python 3.6.15是能同时兼容PyTorch 1.10+和原代码的最高Python版本。新版本P
