[논문 리뷰] MAGE: Nearly Zero-Cost Virtual Memory for Secure Computation

Sam Kumar, David Culler외 1명|arXiv (Cornell University)|2021. 06. 23.

Cryptography and Data Security인용 수 4

한 줄 요약

MAGE는 보안 계산(Secure Computation, SC)의 본질적 무관성( obliviousness)을 활용하여 메모리 액세스 패턴을 사전에 계산하고 최적의 프로액티브 메모리 관리 계획을 수립하는 혁신적인 메모리 프로그래밍 기법을 제안한다. 비동기 프리패칭과 함께 벨라디의 MIN 알고리즘을 적용함으로써, 물리적 메모리보다 큰 작업을 수행하는 SC 워크로드에 대해 메모리 내 성능에 근접한 성능을 달성하며, 운영체제의 가상 메모리보다 최대 10배 빠른 성능을 보인다.

ABSTRACT

Secure Computation (SC) is a family of cryptographic primitives for computing on encrypted data in single-party and multi-party settings. SC is being increasingly adopted by industry for a variety of applications. A significant obstacle to using SC for practical applications is the memory overhead of the underlying cryptography. We develop MAGE, an execution engine for SC that efficiently runs SC computations that do not fit in memory. We observe that, due to their intended security guarantees, SC schemes are inherently oblivious -- their memory access patterns are independent of the input data. Using this property, MAGE calculates the memory access pattern ahead of time and uses it to produce a memory management plan. This formulation of memory management, which we call memory programming, is a generalization of paging that allows MAGE to provide a highly efficient virtual memory abstraction for SC. MAGE outperforms the OS virtual memory system by up to an order of magnitude, and in many cases, runs SC computations that do not fit in memory at nearly the same speed as if the underlying machines had unbounded physical memory to fit the entire computation.

연구 동기 및 목표

• 물리적 메모리보다 큰 프로그램이 실행될 경우 발생하는 높은 메모리 오버헤드와 비효율적인 운영체제 가상 메모리로 인한 보안 계산(SC)의 심각한 성능 저하 문제를 해결하기 위해.
• 메모리 액세스 패턴이 입력 데이터와 무관한 보안 계산 프로토콜의 본질적 무관성을 활용하여 프로액티브이고 사전에 계산된 메모리 관리 전략을 가능하게 하기 위해.
• 물리적 메모리에 맞지 않는 SC 워크로드에 대해 거의 메모리 내 성능에 가까운 성능을 달성함으로써, 거의 비용이 들지 않는 가상 메모리 기능을 실현하기 위해.
• 헤우리스틱 기반 페이지 교체 기법(예: LRU, LFU)의 한계를 극복하기 위해, 이론적으로 최적의 알고리즘인 벨라디의 MIN 알고리즘을 사전 계획 환경에서 적용하기 위해.

제안 방법

• MAGE는 SC 프로토콜이 본질적으로 무관하므로, 메모리 액세스 패턴이 데이터에 의존하지 않으며 실행 이전에 정적 분석이 가능하다는 점을 관찰한다.
• SC 회로를 표현하기 위해 도메인 특화 언어(DSL)를 사용하고, 모든 프로그램 단계에 걸친 완전한 메모리 액세스 패턴을 정적 추출한다.
• 이러한 완전한 액세스 패턴을 바탕으로 MAGE는 벨라디의 MIN 알고리즘을 적용하여 총 메모리 이동 수를 최소화하는 최적의 페이지 교체 순서를 결정한다.
• 사전에 계산된 액세스 패턴을 기반으로 비동기 프리패칭을 구현하여 런타임 동안 데이터 확보로 인한 정지가 발생하지 않도록 보장한다.
• 대규모 메모리 프로그래밍 문제를 효율적으로 해결하기 위해 단계적 계획 접근법을 사용하며, 데이터플로우 그래프를 굵게 다듬음으로써 지수 시간 해법을 피한다.
• SC 프레임워크와 통합하기 위해 런타임에 실행되는, 메모리와 스토리지 간의 사전 계획된 데이터 이동 순서로 구성된 메모리 프로그램을 생성한다.

실험 결과

연구 질문

• RQ1 가용 물리 메모리보다 큰 워킹 세트를 가진 보안 계산(SC) 워크로드가 효율적으로 실행될 수 있는가?
• RQ2 SC 프로토콜의 본질적 무관성을 활용하여 액세스 패턴을 사전에 계산하고 프로액티브 메모리 관리를 가능하게 할 수 있는가?
• RQ3 이론적으로 최적인 벨라디의 MIN 알고리즘 기반 메모리 관리 전략이, 그의 클레어보이언트 성격으로 인해 고전적 페이지 교체에서는 실현 불가능하나, SC 환경에서는 실제로 적용 가능한가?
• RQ4 기존 운영체제 가상 메모리 대비 메모리 프로그래밍 기법이 SC 워크로드에서 런타임 지연을 얼마나 줄이고 처리량을 얼마나 향상시킬 수 있는가?

주요 결과

• 물리 메모리 용량을 초과하는 SC 워크로드에 대해 MAGE는 운영체제 가상 메모리 시스템 대비 최대 10배 빠른 실행 속도를 달성한다.
• MAGE는 일부 SC 계산을 물리 메모리가 무한대일 것처럼 거의 메모리 내 성능으로 실행할 수 있도록 하며, 이는 워킹 세트가 주 메모리 용량을 초과하더라도 가능하다.
• 액세스 패턴을 사전 계산하고 MIN 기반 교체 및 비동기 프리패칭을 통해 런타임 동안 메모리 액세스로 인한 정지를 완전히 제거함으로써, 계산 자원의 고도로 활용 가능한 상태를 유지한다.
• 메모리 프로그래밍 기법 덕분에 MAGE는 이론적으로 최적의 알고리즘인 벨라디의 MIN을 사용할 수 있으며, 이는 고전적 페이지 교체에서는 클레어보이언트 성격으로 인해 실현 불가능하지만, 정적 분석 덕분에 이곳에선 실현 가능해진다.