准确率达100%，「人机交互」机器学习，驱动有机反应精确原子映射研究
编辑|X原子到原子映射（Atom-to-atomMapping，AAM）是识别化学反应前后分子中每个原子位置的任务，这对于理解反应机理非常重要。近年来，随着越来越多的机器学习模型用于逆合成和反应响应结果预测，这些模型的质量高度依赖于反应数据集中AAM的质量。虽然有一些算法使用图论或无监督学习来标记反应数据集的AAM，但现有方法基于子结构对齐而不是化学知识来映射原子。因此，基于子结构对齐的方法可能无法很好地反映原子之间的相互作用。在此，来自韩国首尔大学（SeoulNationalUniversity）和韩国
编辑 | X
原子到原子映射（Atom-to-atom Mapping，AAM）是识别化学反应前后分子中每个原子位置的任务，这对于理解反应机理非常重要。
近年来，随着越来越多的机器学习模型用于逆合成和反应响应结果预测，这些模型的质量高度依赖于反应数据集中 AAM 的质量。虽然有一些算法使用图论或无监督学习来标记反应数据集的 AAM，但现有方法基于子结构对齐而不是化学知识来映射原子。因此，基于子结构对齐的方法可能无法很好地反映原子之间的相互作用。
在此，来自韩国首尔大学（Seoul National University）和韩国科学技术院（KAIST）的研究团队，提出了一种 ML 模型——LocalMapper，可以通过人机回圈（human-in-the-loop）机器学习从化学家标记的反应中学习。这种 ML 模型是针对化学家标记反应过程中的不确定性进行有效处理的一种反应学习算法，可以在学习中不断优化。该算法被认为是一种改进版的自适应积分移动方法（AAM），可以有效提高学习的准确性。
研究表明，通过仅从整个数据集中2%的人类标记反应中学习，就能以98.5%的校准精度预测50K反应的AAM。更重要的是，LocalMapper给出的可信预测覆盖了50 K反应中的97%，对3,000个随机采样的反应显示出100%的准确率。
在分布外（Out-of-distribution，OOD）实验中，发现LocalMapper+性能优于其他现有方法。研究人员期望LocalMapper 可用于生成更精确的反应 AAM，并提高未来基于 ML 的反应预测模型的质量。
相关研究以《Precise atom-to-atom mapping for organic reactions via human-in-the-loop machine learning》为题，于 2024 年 3 月 13 日发布在《Nature Communications》上。
AAM 在化学研究中的重要性
AAM是一种通过识别反应物原子和产物原子之间的一对一映射，准备反应数据中发挥着至关重要的作用。高质量的AAM可以快速识别给定化学反应的反应中心，这对于许多已开发的化学反应分析和预测方法至关重要。
AAM 广泛使用的应用之一是构建缩合反应图 (CGR)。此外，AAM 还可以自动识别反应中心并从数据库中提取反应模板，用于预测反应结果和单步逆合成机器学习模型。由于这些应用高度依赖于反应数据的 AAM，因此 AAM 的质量极大地影响机器学习模型的性能。随着下游模型数量的不断增加，为反应数据集构建高质量的 AAM 成为确保反应预测模型质量的紧迫任务。
现有的 AAM 识别方法通常可分为基于规则的和基于 ML 的方法。尽管比以前的方法准确度更高，但 AAM 需要 100% 的完美准确度，因为反应数据中的缺陷将在下游反应预测模型中被放大。然而，目前还没有可靠的方法来检测可能错误预测的 AAM，这使得预测中的错误难以识别。
此外，尽管现有的基于 ML 的无监督方法比基于规则的方法要快得多，并且适用于更广泛的反应，但在不知道正确的 AAM 的情况下训练模型可能会导致意外错误，即使对于简单的反应也是如此。
三大重要突破
在此，研究人员通过 human-in-the-loop 机器学习提出了一种精确的基于图的 AAM 模型，名为 LocalMapper。
该研究的重要突破体现在以下三个方面：
所提出的基于知识的不确定性识别允许对 ML 模型预测进行快速化学感知验证，为 3,000 个随机采样的置信预测生成 100% 正确的 AAM。 开发的模型 LocalMapper 通过从人机循环机器学习生成的高质量训练数据中学习经化学家验证的 AAM，实现了最先进的 AAM 预测精度。与现有的基于 ML 的模型 RXNMapper 和 GraphormerMapper 相比，仅标记 2% 的反应，显示出更好的预测精度。 在分布外实验中，LocalMapper 比两个现有的基于 ML 的 AAM 模型显示出良好的预测精度，同时保持置信预测的 100% 准确度。
人机循环机器学习框架
为了训练 LocalMapper，研究人员手动标记每个反应的 AAM，以保证训练模型的反应中 AAM 的正确性。由于手动标记化学反应的 AAM 非常耗时（每个反应通常超过一分钟），因此在大型数据集中标记大部分反应是不切实际的。因此，引入主动学习来仅标记一小部分代表性反应。
整个工作流程可以分解为以下 5 个步骤：
随机采样：为了初始化主动学习过程，从未映射的反应数据集中随机采样 k 个反应，其中 k 是人类专家一次性标记 AAM 的一个可承受的小数字。
标签和训练：接下来，手动标记采样的 k 个反应的 AAM，并使用这些反应来训练所提出的基于图的模型 LocalMapper，其结构类似于逆合成模型 LocalRetro 和反应结果预测模型 LocalTransform。从人类绘制的反应中提取的反应模板用于更新模板库，该模板库将用于后续的不确定性识别。
AAM 预测：接下来，使用 LocalMapper 来预测数据集中所有反应的反应物和产物之间的原子相关性。根据 LocalMapper 预测的原子-原子相关性，按照 Schwaller 等人引入的原子映射程序生成每个反应的 AAM。
置信度识别：对于每个预测反应的 AAM，提取反应模板来表示其反应模式。如果提取的反应模板存在于当前模板库中，则该反应预测的 AAM 集合被认为是置信预测，否则是不确定预测。
主动采样：对于从不确定预测中提取的每个唯一模板，从共享最多反应的模板开始对一个反应进行采样，直到采样到 k 个反应。然后，这些反应由人类化学家标记，并在下一次迭代中用于训练模型，重复步骤 2。
从第二次迭代开始，研究人员使用半监督学习来训练模型，从每个唯一验证的反应模板的置信预测中采样 100 个反应，以提高模型的稳健性。这些采样反应按 9:1 的比例分为训练集和验证集，以防止过度拟合。
LocalMapper
为了预测反应中反应物和产物之间的 AAM，研究人员设计了一个基于图的模型，称为 LocalMapper，以了解反应物中的每个原子重新定位到产物中的原子的概率。
与之前的逆合成模型 LocalRetro 和反应结果预测 LocalTransform 类似，使用图来表示分子，以原子为节点，键为边，并通过反应中原子的局部和全局特征来学习 AAM 通过消息传递神经网络和注意力机制。
首先，研究人员使用 3 个消息传递层对每个原子的局部化学环境进行编码，并通过 3 个多头交叉注意块根据反应物的原子特征更新产物中的原子特征。在反应物和产物之间的每个原子的特征充分传达后，通过单头注意力块计算产物和反应物之间的 AAM 相关性。
使用 Softmax 函数对注意力分数进行归一化后，估计反应物中的每个原子与产物中的每个原子是相同原子的概率。按照 RXNMapper 中引入的原子映射程序，使用生成的概率从最高概率到最低概率识别从产物到反应物的 AAM。
100% 预测准确率
总之，研究人员提出了一种基于图的 ML 模型 LocalMapper，以通过人机循环机器学习精确识别大型反应数据集的 AAM。通过利用专业知识手动标记少量反应数据，训练了人机循环机器学习模型，以精确、自动地标记大量具有相似反应规则的反应。
表 1：手动检查反应 AAM 前后，RXNMapper、GraphormerMapper 和 LocalMapper 在 USPTO-50K 数据集上的 AMM 结果。（来源：论文）
对于公开可用的 USPTO-50K 数据集，该模型仅通过学习 2% 的化学家标记反应，就能以 98.5% 的准确率预测 AAM。
表 2：RXNMapper、GraphormerMapper 和 LocalMapper 在四个不同源上检查的手动映射反应的 AMM 结果。（来源：论文）
更重要的是，LocalMapper 自信预测的数据集中 97% 的反应的 AAM，表现出 100% 的预测准确率。并且在不同的分布外测试集中也观察到类似的结果。
研究人员期望所提出的 LocalMapper 可用于为未来的下游反应预测模型提供精确的反应 AAM，并有利于化学界了解更多有关反应数据集的统计见解。
GitHub 地址：https://github.com/snu-micc/LocalMapper
注：封面来自网络
WPS Office是一款面向macOS用户的综合办公套件，将文字处理、电子表格、演示文稿、PDF工具、模板与云协作整合在同一应用中。它兼容常见Microsoft Office格式，支持Intel及Apple芯片Mac，基础功能可免费使用，
Shapr3D是一款面向工业设计、机械工程、建筑概念和三维打印工作流的CAD软件。Mac版采用Parasolid建模内核，支持草图约束、实体建模、工程图、可视化渲染及常见CAD格式交换，并可通过账户在多台设备之间同步项目。
REAPER是Cockos开发的数字音频工作站，提供多轨音频与MIDI录制、剪辑、处理、混音和母带制作工具。Mac版兼容Intel与Apple芯片，支持AU、VST、VST3、CLAP等插件格式，并提供高度可定制的工作流程。
Ableton Live 是面向音乐制作人与现场表演者的数字音频工作站，提供编曲视图、独具特色的现场视图、音频录制、MIDI创作、实时变速、乐器及效果器。Mac版原生支持Apple芯片，并可连接音频接口、MIDI控制器和第三方插件。
Adobe After Effects 是面向动态图形、影视特效与视频合成的专业创作软件。它提供关键帧动画、遮罩抠像、运动跟踪、三维空间、文字动画和表达式等工具，并可与 Premiere Pro、Photoshop、Illustrator
Adobe Audition 是面向播客、影视后期、音乐制作及内容创作者的专业音频工作站，提供波形精修、多轨录制与混音、频谱编辑、降噪修复、响度匹配和效果处理工具，并可与 Adobe Premiere 协同完成视频声音制作。
Adobe Bridge 是面向摄影师、设计师及内容团队的免费数字资产管理工具，可集中预览照片、视频和 Adobe 项目文件，并利用评级、标签、关键词、元数据、筛选与收藏集完成分类检索。它还支持批量重命名、格式导出、相机导入及 Camera
Adobe Illustrator 是 Adobe 推出的专业矢量图形设计软件，适合在 Mac 上制作标志、图标、插画、包装、信息图及印刷版面。它以路径和锚点构建可无损缩放的图形，并提供钢笔、形状、文字、图像描摹、渐变、画板及生成式功能。
Adobe InDesign 是面向印刷品与数字出版物的专业版面设计软件，可在 Mac 上制作书籍、杂志、宣传册、海报、报告及交互式文档。它提供精细的文字样式、网格、母版、长文档管理和印前输出能力。
Adobe Lightroom Classic 是面向桌面摄影工作流的照片管理与后期处理软件，可将原始照片保存在本地硬盘，通过目录、关键词、评分和收藏夹高效整理图库，并提供 RAW 冲印、蒙版、镜头校正、批量同步及多格式导出等功能。
Adobe Media Encoder 是 Adobe 推出的专业音视频编码与转码工具，可通过队列、预设和监视文件夹批量完成格式转换、代理文件创建及多平台成片输出，并与 Premiere Pro、After Effects 等软件紧密协作。
Adobe Photoshop 是面向摄影师、设计师和内容创作者的专业图像处理软件，提供图层、蒙版、选区、修复、调色、文字排版、智能对象及生成式编辑等能力。Mac 版支持 Intel 与 Apple Silicon 处理器。
Adobe Photoshop 是面向摄影师、设计师和内容创作者的专业图像处理工具，可在 Windows 上完成照片修饰、选区抠图、图层合成、文字排版、绘画及生成式编辑。软件采用订阅授权，由 Creative Cloud 管理安装、更新与许
Adobe InDesign 是面向印刷与数字出版的专业页面排版工具，可制作书籍、杂志、宣传册、海报、交互式 PDF 和电子书。它提供精细的文字样式、多页面管理、印前检查及多格式输出能力，并能与 Photoshop、Illustrator
Adobe Illustrator 是用于绘制矢量图形、插画、图标、徽标和版式的专业设计工具。它以路径和锚点为核心，作品放大后仍能保持清晰，并提供画板、文字、渐变、图像描摹及精确变换等功能，适合品牌视觉、印刷物料、数字界面和商业插画创作。
Adobe Bridge 是 Adobe 推出的创意资产管理工具，可集中预览、分类和查找照片、视频、PDF及多种设计文件，并支持元数据、关键词、评级、批量重命名和格式导出。它可与 Photoshop、Illustrator 等应用衔接，
Affinity Photo 是一款面向摄影师、修图师和视觉设计人员的桌面图像编辑工具，提供 RAW 显影、多图层合成、蒙版、实时滤镜、批处理和色彩管理等功能。原独立产品已停止销售，其照片编辑能力已延续至免费的新 Affinity 应用；
Adobe Lightroom Classic 是面向 Windows 桌面摄影工作流的照片管理与无损编辑工具，可通过目录集中整理大型图库，并完成 RAW 显影、局部蒙版、批量调整、打印和多格式导出。它强调本地文件管理与精细控制，
Figma 是面向产品设计与开发团队的云端协作平台，可在 Windows 桌面客户端中完成界面绘制、组件管理、交互原型、评论评审和开发交付。文件保存在云端，多名成员能够同步编辑并共享设计系统；官方同时提供免费 Starter 方案及按席位计
Affinity Designer 是面向矢量图形、数字插画和界面设计的专业创作工具，可在矢量、像素与导出工作区之间快速切换。它提供曲线编辑、形状构建、画板、排版、图层和专业色彩管理，并支持常见设计文件的导入及多种印刷、网页格式输出。
Capture One Pro 是面向摄影师和影像团队的专业桌面照片处理工具，提供相机定制色彩配置、RAW转换、图层与蒙版、联机拍摄、目录及会话管理等功能。它适合棚拍、人像、商业产品、婚礼和风光摄影，
Adobe Acrobat Pro 是面向专业 PDF 工作的综合解决方案，可在 Windows 中创建、编辑、转换、比较、整理和保护文档，并支持表单、电子签名、扫描件文字识别及敏感信息永久遮盖。订阅还包含网页、移动服务和云端协作能力，
Adobe After Effects 是面向动态图形、动画与影视视觉特效制作的专业工具，可完成关键帧动画、图层合成、遮罩跟踪、抠像、文字动画和二维三维元素整合。Windows 版通过 Adobe Creative Cloud 安装及更新，
DaVinci Resolve是Blackmagic Design推出的一体化影视后期工具，将素材管理、非线性剪辑、专业调色、Fusion视觉特效、Fairlight音频处理和成片交付整合在统一工程中。免费版已覆盖多数个人创作需求，