r/MachineLearning에서 주목한 Flower: 워프 기반 Neural PDE Solver

게시글 개요

r/MachineLearning의 [R] Neural PDE solvers built (almost) purely from learned warps는 2026-02-23 UTC 기준 점수 79, 댓글 20을 기록했다. 작성자가 “자신의 연구”임을 명시하고, ResearchGate 논문 및 GitHub 코드를 함께 공개한 형태다.

제안된 접근: spatial warp를 중심 연산으로

게시글 설명에 따르면 Flower는 Fourier layer, attention, 대규모 spatial convolution에 의존하지 않고, 각 위치에서 변위를 예측해 이동된 좌표의 특징을 샘플링하는 방식으로 공간 상호작용을 만든다. 멀티헤드, value projection, skip connection, U-Net 스캐폴드를 결합했지만, 스케일 내부의 핵심 상호작용은 워프 연산에 집중된 구조다.

작성자는 이 방식이 격자 포인트 수에 선형적으로 대응해 3D 문제에서도 효율적일 수 있다고 설명했다. 즉, 고비용 연산 블록 대신 데이터 이동 방향을 학습하는 방식으로 PDE 동역학을 근사한다는 주장이다.

게시글에서 제시한 성능 주장

• The Well 중심 16개 데이터셋에서 동급 파라미터(약 15-20M) 비교 시 one-step 예측 우위
• 20-step autoregressive rollout에서도 대체로 우수하나 일부 과제는 모든 모델이 어려움
• 150M 스케일 모델이 특정 compressible Euler 설정에서 대형 사전학습 모델(Poseidon 628M) 대비 우수했다고 보고

한계와 커뮤니티 피드백

작성자 본문은 장기 rollout에서 이점이 줄어드는 경향, 특정 조건의 안정성 이슈를 함께 언급했다. 댓글에서도 “기상 예측처럼 더 어려운 PDE에 확장 가능한가”, “충격파/불연속 상황에서 warp 기반 접근이 얼마나 견고한가” 같은 질문이 나왔다. The Well 관련 참여자 계정에서도 흥미로운 결과라는 반응이 확인됐다.

현재 이 결과는 작성자 보고 중심이며 arXiv 공개 전 단계라는 점을 함께 보는 것이 합리적이다. 그럼에도 구조적 단순성으로 높은 성능을 노리는 시도가 scientific ML 커뮤니티에 의미 있는 실험 축을 던졌다는 점은 분명하다.

출처: Reddit 스레드, ResearchGate 논문 링크, GitHub 코드