HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Diffusion Unit: Interpretable Edge Enhancement and Suppression Learning for 3D Point Cloud Segmentation

Xiu Haoyi ; Liu Xin ; Wang Weimin ; Kim Kyoung-Sook ; Shinohara Takayuki ; Chang Qiong ; Matsuoka Masashi

Diffusion Unit: Interpretable Edge Enhancement and Suppression Learning
  for 3D Point Cloud Segmentation

Abstract

3D point clouds are discrete samples of continuous surfaces which can be usedfor various applications. However, the lack of true connectivity information,i.e., edge information, makes point cloud recognition challenging. Recentedge-aware methods incorporate edge modeling into network designs to betterdescribe local structures. Although these methods show that incorporating edgeinformation is beneficial, how edge information helps remains unclear, makingit difficult for users to analyze its usefulness. To shed light on this issue,in this study, we propose a new algorithm called Diffusion Unit (DU) thathandles edge information in a principled and interpretable manner whileproviding decent improvement. First, we theoretically show that DU learns toperform task-beneficial edge enhancement and suppression. Second, weexperimentally observe and verify the edge enhancement and suppressionbehavior. Third, we empirically demonstrate that this behavior contributes toperformance improvement. Extensive experiments and analyses performed onchallenging benchmarks verify the effectiveness of DU. Specifically, our methodachieves state-of-the-art performance in object part segmentation usingShapeNet part and scene segmentation using S3DIS. Our source code is availableat https://github.com/martianxiu/DiffusionUnit.

Code Repositories

martianxiu/diffusionunit
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
3d-part-segmentation-on-shapenet-partDiffusion Unit
Class Average IoU: 85.2
Instance Average IoU: 87.1
semantic-segmentation-on-s3dis-area5Diffusion Unit
Number of params: 8.1M
mAcc: 78.1
mIoU: 72.2
oAcc: 91.3

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Diffusion Unit: Interpretable Edge Enhancement and Suppression Learning for 3D Point Cloud Segmentation | Papers | HyperAI