Monge-Ampere Regularization for Learning Arbitrary Shapes from Point Clouds

TPAMI 2025

1Shandong University, 2City University of Hong Kong, 3Nanyang Technological University, 4Texas A&M University

Abstract

As commonly used implicit geometry representations, the signed distance function (SDF) is limited to modeling watertight shapes, while the unsigned distance function (UDF) is capable of representing various surfaces. However, its inherent theoretical shortcoming, i.e., the non-differentiability at the zero level set, would result in sub-optimal reconstruction quality.

In this paper, we propose the scaled-squared distance function (S2DF), a novel implicit surface representation for modeling arbitrary surface types. S2DF does not distinguish between inside and outside regions while effectively addressing the non-differentiability issue of UDF at the zero level set. We demonstrate that S2DF satisfies a second-order partial differential equation of Monge-Ampere-type, allowing us to develop a learning pipeline that leverages a novel Monge-Ampere regularization to directly learn S2DF from raw unoriented point clouds without supervision from ground-truth S2DF values. Extensive experiments across multiple datasets show that our method significantly outperforms state-of-the-art supervised approaches that require ground-truth surface information as supervision for training.

Method

S2DF

We introduce scaled-squared distance function (S2DF), a novel implicit surface representation capable of representing shapes of arbitrary types while remaining differentiable at the zero level set.


Monge-Ampere Regularization

We analyze the theoretical properties of S2DF, demonstrating that it satisfies a Monge-Ampere equation. Based on this, we propose Monge-Ampere regularization to directly learn S2DF from raw unoriented point clouds without supervision from ground-truth S2DF values.

Comparison Results

MGN

3D Scene

Thingi10K

Shapenet

Stanford 3D Scanning Repository

LevelSetUDF

Ours

LevelSetUDF

Ours

Waymo

More Results

BibTeX

@ARTICLE{yang2025monge,
  author={Yang, Chuanxiang and Zhou, Yuanfeng and Wei, Guangshun and Ma, Long and Hou, Junhui and Liu, Yuan and Wang, Wenping},
  journal={IEEE Transactions on Pattern Analysis and Machine Intelligence}, 
  title={Monge-Ampere Regularization for Learning Arbitrary Shapes from Point Clouds}, 
  year={2025},
  volume={},
  number={},
  pages={1-15},
  keywords={Surface reconstruction;Point cloud compression;Shape;Neural networks;Training;Mathematical models;Three-dimensional displays;Decoding;Topology;Surface fitting;Implicit neural representation;distance function;surface reconstruction},
  doi={10.1109/TPAMI.2025.3563601}
}