Graph curvature method enhances neural network pruning

Researchers have introduced a novel approach to neural network pruning by leveraging graph theory, specifically Ollivier-Ricci curvature (ORC). This method identifies critical data flows and connections within a neural network by analyzing activation patterns and calculating a metric called neural curvature (NC). Experiments on image datasets like MNIST and CIFAR-100 demonstrate that this NC-based pruning can more effectively identify unimportant edges compared to existing techniques. AI

IMPACT Introduces a novel graph-theory-based method for optimizing neural networks, potentially leading to more efficient models.

RANK_REASON The cluster contains a research paper detailing a new method for neural network pruning. [lever_c_demoted from research: ic=1 ai=1.0]

Read on arXiv cs.LG →

• CIFAR-10
• CIFAR-100
• MNIST
• Shuhang Tan

• paper
• other

AI-generated summary · Google Gemini · from 1 sources. How we write summaries →