ARC-DSL: 一种用于抽象和推理语料库的领域特定语言

ARC-DSL:为抽象和推理设计的领域特定语言

抽象和推理语料库(Abstraction and Reasoning Corpus, ARC)是一个旨在评估人工智能系统抽象推理能力的基准测试集。为了更好地解决ARC任务,研究人员开发了ARC-DSL(Domain Specific Language for ARC),这是一种专门为ARC设计的领域特定语言。本文将深入探讨ARC-DSL的设计理念、核心功能以及使用示例,帮助读者全面了解这一创新工具。

ARC-DSL的设计理念

ARC-DSL的设计初衷是创建一种既具有足够表达能力又保持通用性的编程语言。它的目标是:

能够编写程序解决任意ARC任务
仅包含少量原语(primitives),但每个原语都适用于多种任务

这种设计理念使得ARC-DSL在保持简洁的同时,也具备了强大的问题解决能力。

ARC-DSL的核心组件

ARC-DSL的核心组件包括:

dsl.py: 定义了ARC-DSL的基本原语和操作。
solvers.py: 包含了使用ARC-DSL编写的训练任务求解程序。
arc_dsl_writeup.pdf: 提供了对ARC-DSL工作的详细描述。

这些组件共同构成了ARC-DSL的基础架构,为用户提供了完整的开发环境。

ARC-DSL使用示例

为了更好地理解ARC-DSL的实际应用,我们来看两个具体的求解器示例:

示例1: 任务00d62c1b的求解器

Task 00d62c1b

def solve_00d62c1b(I):
    objs = objects(grid=I, univalued=T, diagonal=F, without_bg=F)
    black_objs = colorfilter(objs=objs, value=ZERO)
    borders = rbind(function=bordering, fixed=I)
    does_not_border = compose(outer=flip, inner=borders)
    enclosed = mfilter(container=black_objs, function=does_not_border)
    O = fill(grid=I, value=FOUR, patch=enclosed)
    return O

这个求解器的工作流程如下:

从输入网格中提取单色对象
筛选出黑色对象
创建一个判断对象是否接触网格边界的函数
找出不接触边界的黑色对象(即"洞")
将这些"洞"填充为黄色,得到输出网格

示例2: 任务5521c0d9的求解器

Task 5521c0d9

def solve_5521c0d9(I):
    objs = objects(grid=I, univalued=T, diagonal=F, without_bgT)
    foreground = merge(containers=objs)
    empty_grid = cover(grid=I, patch=foreground)
    offset_getter = chain(h=toivec, g=invert, f=height)
    shifter = fork(outer=shift, a=identity, b=offset_getter)
    shifted = mapply(function=shifter, container=objs)
    O = paint(grid=empty_grid, obj=shifted)
    return O

这个求解器的工作流程如下: