11.1SIJun 2
Ollivier-Ricci curvature in cycle overlap modeZexian Zhou, Bo Jiao
Ollivier-Ricci curvature of an edge (x,y) is defined by comparing the distance taken to transport from neighbors of x to neighbors of y. It is a structural measure that has been studied in many fields such as community detection and deep neural networks. However, high computational complexity or error limits its application in large scale-free graphs. This paper proposes an optimal transport principle to minimize the distance by 3,4,5-cycles that include the edge (x,y), and designs a curvature calculation approach named Curvature in Cycle Overlap Mode (CCOM). In this approach, a greedy and pruning algorithm is proposed to approximate the optimal transport principle. We theoretically and experimentally verified that our approach CCOM can significantly improve the accuracy of the curvature on real-world networks with low time consumption. In addition, we compared CCOM with baseline approximation approaches in community detection tasks using the same curvature-based framework, and experimentally confirmed the effectiveness of CCOM on large scale-free graphs.
CVJul 16, 2022Code
CA-SpaceNet: Counterfactual Analysis for 6D Pose Estimation in SpaceShunli Wang, Shuaibing Wang, Bo Jiao et al.
Reliable and stable 6D pose estimation of uncooperative space objects plays an essential role in on-orbit servicing and debris removal missions. Considering that the pose estimator is sensitive to background interference, this paper proposes a counterfactual analysis framework named CASpaceNet to complete robust 6D pose estimation of the spaceborne targets under complicated background. Specifically, conventional methods are adopted to extract the features of the whole image in the factual case. In the counterfactual case, a non-existent image without the target but only the background is imagined. Side effect caused by background interference is reduced by counterfactual analysis, which leads to unbiased prediction in final results. In addition, we also carry out lowbit-width quantization for CA-SpaceNet and deploy part of the framework to a Processing-In-Memory (PIM) accelerator on FPGA. Qualitative and quantitative results demonstrate the effectiveness and efficiency of our proposed method. To our best knowledge, this paper applies causal inference and network quantization to the 6D pose estimation of space-borne targets for the first time. The code is available at https://github.com/Shunli-Wang/CA-SpaceNet.
LGMar 2, 2025Code
Hierarchical graph sampling based minibatch learning with chain preservation and variance reductionQia Hu, Bo Jiao
Graph sampling-based Graph Convolutional Networks (GCNs) decouple sampling from forward and backward propagation during minibatch training, enhancing scalability with respect to layer depth and graph size. We propose HIS_GCNs, a hierarchical importance sampling-based learning method. By constructing minibatches using sampled subgraphs, HIS_GCNs focuses on the importance of both the core and periphery in a scale-free training graph. Specifically, it preserves the centrum of the core in most minibatches, which maintains connectivity between periphery nodes, and samples periphery edges without core node interference, which allows longer chains composed entirely of low-degree nodes remain within the same minibatch. HIS_GCNs can maximize the discrete Ricci curvature (i.e., Ollivier-Ricci curvatures) of the edges in a subgraph, enabling preservation of important chains for information propagation. This approach can achieve a low node embedding variance and a high convergence speed. Diverse experiments on Graph Neural Networks (GNNs) with node classification tasks confirmed the superior performance of HIS_GCNs in terms of both accuracy and training time. Open-source code (https://github.com/HuQiaCHN/HIS-GCN).