NSFlow: Neuro-Symbolic AI를 위한 확장 가능한 FPGA 프레임워크
NSFlow는 FPGA 기반의 Neuro-Symbolic AI 가속화 프레임워크로, 기존 하드웨어의 한계를 극복하고 뛰어난 성능과 확장성을 제공합니다. 다양한 NSAI 작업에서 압도적인 속도 향상을 달성하여 차세대 인지 시스템 개발에 중요한 역할을 할 것으로 기대됩니다.

AI의 새로운 지평을 여는 NSFlow
최근 급격한 발전을 거듭하고 있는 인공지능(AI) 분야에서, Neuro-Symbolic AI (NSAI) 는 주목받는 새로운 패러다임입니다. NSAI는 신경망과 기호적 추론을 통합하여 AI 시스템의 투명성, 추론 능력, 데이터 효율성을 향상시키는 것을 목표로 합니다. 특히, 추론 작업과 인간-AI 협업 시나리오에서 뛰어난 성능을 보여주면서 주목받고 있습니다.
하지만 기존의 CPU, GPU, TPU와 같은 하드웨어에서 NSAI 작업을 실행하는 것은 쉽지 않습니다. 이기종 컴퓨팅 커널, 높은 메모리 사용량, 독특한 메모리 접근 패턴 등이 그 이유입니다. 게다가 현재의 NSAI 알고리즘은 작동 유형과 규모가 다양하여 기존의 머신러닝 가속기에 적용하기 어렵습니다.
이러한 문제점을 해결하기 위해, Yang Hanchen 등 연구진은 FPGA 기반의 가속화 프레임워크인 NSFlow를 개발했습니다. NSFlow는 NSAI 시스템 전반에서 높은 효율성, 확장성, 다양성을 달성하도록 설계되었습니다.
NSFlow의 핵심은 무엇일까요?
NSFlow는 작업량의 데이터 의존성을 식별하고 최적화된 데이터 흐름 아키텍처를 생성하는 설계 아키텍처 생성기를 갖추고 있습니다. 또한, 유연한 연산 유닛, 재구성 가능한 메모리, 혼합 정밀도 기능을 갖춘 재구성 가능한 어레이를 통해 다양한 NSAI 작업에 효율적으로 대응할 수 있습니다.
실제 성능은 어떨까요?
다양한 NSAI 작업에 대한 평가 결과, NSFlow는 Jetson TX2 대비 31배, GPU 대비 2배 이상, TPU 유사 시스토릭 어레이 대비 8배, Xilinx DPU 대비 3배 이상의 속도 향상을 달성했습니다. 뿐만 아니라, 심볼릭 작업량이 150배 증가하더라도 실행 시간은 단 4배만 증가하는 뛰어난 확장성을 보여주었습니다.
결론적으로, NSFlow는 실시간으로 일반화 가능한 NSAI 알고리즘 가속화를 가능하게 하는 최초의 프레임워크로, 차세대 인지 시스템에 대한 유망한 해결책을 제시합니다. NSAI 분야의 발전과 함께 NSFlow의 역할 또한 더욱 중요해질 것으로 예상됩니다. 이는 단순한 기술적 진보를 넘어, 더욱 지능적이고 효율적인 AI 시스템 개발을 위한 중요한 이정표가 될 것입니다.
Reference
[arxiv] NSFlow: An End-to-End FPGA Framework with Scalable Dataflow Architecture for Neuro-Symbolic AI
Published: (Updated: )
Author: Hanchen Yang, Zishen Wan, Ritik Raj, Joongun Park, Ziwei Li, Ananda Samajdar, Arijit Raychowdhury, Tushar Krishna
http://arxiv.org/abs/2504.19323v2