AS2 -- Attention-Based Soft Answer Sets: An End-to-End Differentiable Neuro-Soft-Symbolic Reasoning Architecture

TL;DR

AS2通过软化ASP算子实现端到端可微推理,在Visual Sudoku上达99.89%准确率。

cs.AI 🔴 高级 2026-03-19 4 次浏览
Wael AbdAlmageed
神经符号 可微推理 ASP 视觉数独 机器学习

核心发现

方法论

AS2采用注意力机制和软化的ASP算子,保持符号域的概率分布,通过最小化TP算子的固定点残差实现端到端训练。模型不使用传统位置嵌入,而是通过约束组成员嵌入编码问题结构。

关键结果

  • 在Visual Sudoku上,AS2实现了99.89%的单元格准确率和100%的约束满足率,使用贪婪约束解码,无需外部求解器。
  • 在MNIST加法任务中,AS2在所有规模上实现了超过99.7%的数字准确率。
  • 实验结果表明,软化的可微固定点算子结合约束感知注意力可匹敌或超越现有神经符号系统。

研究意义

AS2在神经符号AI中实现了完全可微的推理架构,消除了传统方法中感知与推理组件之间的非可微边界。该方法不仅提高了推理准确性,还为约束反馈提供了直接的梯度流,从而改善了感知模块的训练。

技术贡献

AS2通过软化ASP算子实现了端到端可微推理,避免了传统方法中对外部求解器的依赖。其约束组成员嵌入设计消除了对位置嵌入的需求,直接反映了ASP规范,使模型对位置索引不敏感。

新颖性

AS2首次在神经符号推理中实现了完全可微的ASP推理,避免了传统方法中感知与推理之间的非可微边界。其创新在于通过软化的TP算子实现约束反馈的梯度流。

局限性

  • AS2在处理更复杂的约束时可能面临计算瓶颈,尤其是在大规模数据集上。
  • 模型对约束组的定义高度依赖,可能限制其在其他任务中的泛化能力。

未来方向

未来工作可探索AS2在更复杂约束和更大规模数据集上的应用,进一步优化其计算效率和泛化能力。

AI 总览摘要

神经符号AI系统通常在神经感知模块和离散符号求解器之间存在非可微边界,阻止了约束满足反馈在训练期间到达感知编码器。AS2(基于注意力的软答案集)是一种完全可微的神经符号架构,通过软化ASP立即后果算子TP替代离散求解器。AS2在整个前向传递过程中保持符号域的概率分布,通过最小化TP的概率提升的固定点残差进行端到端训练,从而在不调用外部求解器的情况下实现约束检查的微分。该架构完全不使用传统位置嵌入,而是通过直接反映声明性ASP规范的约束组成员嵌入编码问题结构,使模型对任意位置索引不敏感。在Visual Sudoku上,AS2在1,000个测试板上实现了99.89%的单元格准确率和100%的约束满足率(由Clingo验证),使用贪婪约束解码程序,无需外部求解器。在MNIST加法任务中,AS2在所有规模上实现了超过99.7%的数字准确率。这些结果表明,软化的可微固定点算子结合约束感知注意力和声明性约束规范,可以匹敌或超越基于管道和求解器的神经符号系统,同时保持完全的端到端可微性。

深度分析

研究背景

神经符号AI近年来受到广泛关注,其目标是结合神经网络的感知能力和符号推理的逻辑一致性。传统方法通常将感知与推理分离,导致训练过程中缺乏约束反馈。

核心问题

神经符号系统中,感知模块与符号求解器之间的非可微边界阻碍了约束反馈的流动,影响了感知模块的训练效果。

核心创新

AS2通过软化ASP算子实现了完全可微的推理,采用约束组成员嵌入消除了位置嵌入的需求,直接编码问题结构。

方法详解

  • �� 使用软化的ASP算子TP替代离散求解器
  • �� 通过约束组成员嵌入编码问题结构
  • �� 采用多层Transformer进行推理
  • �� 使用TP固定点残差作为损失函数进行训练

实验设计

在Visual Sudoku和MNIST加法任务上进行测试,使用贪婪约束解码和多层Transformer架构,评估模型的准确性和约束满足率。

结果分析

AS2在Visual Sudoku上实现了99.89%的单元格准确率和100%的约束满足率,在MNIST加法任务中实现了超过99.7%的数字准确率。

应用场景

AS2可用于需要高精度推理的任务,如视觉数独和复杂算术推理,具有广泛的应用潜力。

局限与展望

AS2在处理更复杂的约束时可能面临计算瓶颈,未来工作可探索其在更大规模数据集上的应用。

通俗解读 非专业人士也能看懂

想象一个工厂,工人们需要按照特定规则组装产品。传统方法中,工人们只负责自己的部分,无法了解整体产品的组装情况。AS2就像一个智能管理系统,实时监控每个工人的工作,确保所有部件都符合最终产品的要求。这样,工厂可以更高效地生产出符合标准的产品,而不需要额外的检查步骤。

简单解释 像给14岁少年讲一样

想象你在玩一个拼图游戏,每块拼图都有特定的位置。传统方法就像你只能看到一块拼图,而不知道整体图案。AS2就像一个超级助手,帮你实时查看每块拼图的位置,确保它们都在正确的位置上。这样,你就能快速完成拼图,而不需要反复调整!

术语表

ASP (答案集编程)

一种用于定义和求解逻辑程序的声明性编程范式。

用于定义AS2中的约束规范。

TP算子

逻辑程序的立即后果算子,用于推导符号域的固定点。

AS2中用于实现可微推理的核心算子。

约束组成员嵌入

一种编码问题结构的方法,通过反映约束组成员关系来替代位置嵌入。

用于AS2中编码问题结构。

Visual Sudoku

一种视觉感知任务,要求模型根据数独规则推理出正确的数字排列。

AS2的主要测试任务之一。

MNIST加法任务

一种基于MNIST数据集的算术推理任务,要求模型预测数字之和。

用于评估AS2的算术推理能力。

开放问题 这项研究留下的未解疑问

  • 1 如何在更复杂的约束条件下保持AS2的计算效率?
  • 2 AS2在处理开放世界推理任务时的表现如何?

应用场景

近期应用

视觉数独求解

AS2可用于高效求解视觉数独问题,适用于教育和游戏领域。

远期愿景

复杂推理系统

AS2有潜力用于开发更复杂的推理系统,如自动驾驶中的实时决策。

原文摘要

Neuro-symbolic artificial intelligence (AI) systems typically couple a neural perception module to a discrete symbolic solver through a non-differentiable boundary, preventing constraint-satisfaction feedback from reaching the perception encoder during training. We introduce AS2 (Attention-Based Soft Answer Sets), a fully differentiable neuro-symbolic architecture that replaces the discrete solver with a soft, continuous approximation of the Answer Set Programming (ASP) immediate consequence operator $T_P$. AS2 maintains per-position probability distributions over a finite symbol domain throughout the forward pass and trains end-to-end by minimizing the fixed-point residual of a probabilistic lift of $T_P$, thereby differentiating through the constraint check without invoking an external solver at either training or inference time. The architecture is entirely free of conventional positional embeddings. Instead, it encodes problem structure through constraint-group membership embeddings that directly reflect the declarative ASP specification, making the model agnostic to arbitrary position indexing. On Visual Sudoku, AS2 achieves 99.89% cell accuracy and 100% constraint satisfaction (verified by Clingo) across 1,000 test boards, using a greedy constrained decoding procedure that requires no external solver. On MNIST Addition with $N \in \{2, 4, 8\}$ addends, AS2 achieves digit accuracy above 99.7% across all scales. These results demonstrate that a soft differentiable fixpoint operator, combined with constraint-aware attention and declarative constraint specification, can match or exceed pipeline and solver-based neuro-symbolic systems while maintaining full end-to-end differentiability.

cs.AI