연구진은 저차원 점 구름에 대한 제곱 유클리드 비용의 대규모 엔트로피 최적 운송(EOT) 문제를 단일 GPU로 해결하는 솔버 'FlashSinkhorn 2(FS2)'를 공개했다. FS2는 셀 중심에서 거친 단계로 포텐셜을 계산한 뒤, 모턴 순서의 블록 희소 미세 단계로 중심 근사 오차만 보정하는 2단계 구조를 사용해 모든 점 쌍을 매번 계산하지 않고도 정확도를 확보한다. 단일 A100 GPU에서 우주론적 N체 시뮬레이션의 1.34억 개 입자 쌍 사이의 EOT 문제를 2.5시간 이내에 전체 입자 기준 잔차 0.01 이하로 풀어내, 이 정확도로 해결된 최대 규모의 이산 EOT 문제임을 보였다. 코드는 깃허브에 공개돼 재현 가능하다.
- •거친 단계(셀 중심)와 모턴 순서 블록 희소 미세 단계를 결합한 2단계 솔버로 전체 점 쌍 평가를 회피
- •32개 합성 벤치마크 모두 목표 잔차 도달, 경쟁 기법 GeomLoss 멀티스케일은 10개만 성공
- •단일 A100에서 1.34억 입자 우주론 시뮬레이션 EOT를 2.5시간 내 잔차 0.01 이하로 해결
- •이 정확도 기준 역대 최대 규모 이산 엔트로피 최적 운송 문제 해결, 오픈소스 공개
0단 자동
AI가 규칙대로 쓰고 그대로 게시했습니다. 사람이 따로 보지 않았습니다.
- 규칙 판
- 규칙 판 도입 이전 기사입니다.
- 남기는 것
- 규칙 판 · 모델 · 시각
- 판 기록
- 아직 없습니다.
FlashSinkhorn 2: Block-Sparse Entropic Optimal Transport
- 1.FlashSinkhorn 2(FS2)는 센트로이드 단위 '코스 단계'와 블록-희소 '파인 단계'를 결합해 대규모 최적 수송 문제를 품
- 2.모든 점 쌍을 계산하던 기존 스트리밍 GPU 솔버와 달리 대부분의 연산을 센트로이드 수준으로 대체
- 3.A100 한 장으로 1.34억 입자짜리 우주론 N-body 시뮬레이션 두 세트 간 최적 수송을 2.5시간 이내에 해결
- 4.32개 합성 벤치마크 전부에서 목표 잔차에 도달했고, 소스코드를 깃허브에 공개
왜 중요한가?
지금까지 이 정확도로 풀린 사례 중 가장 큰 규모의 이산 엔트로피 최적 수송 문제를 단일 GPU로 해결해, 대규모 과학 시뮬레이션 데이터 비교에 실용적인 도구를 제공한다.
언급 프로젝트
본문 미리보기
arXiv:2610.02395v1 Announce Type: new Abstract: Streaming GPU solvers for entropic optimal transport (EOT), such as FlashSinkhorn, avoid storing the dense kernel but still evaluate all $n\times m$ point pairs in every Sinkhorn iteration. We present \textbf{FlashSinkhorn~2} (FS2), a solver for squared-Euclidean cost on low-dimensional point clouds that solves large discrete EOT problems to a prescribed marginal residual on a single GPU by coupling two stages. A coarse stage solves on cell centro
전체 내용이 궁금하다면?
원문을 직접 읽어보세요
이 글이 만들어진 과정
- 10:40AI 초안
