3.1. Problem Statement
The general objective of 3D point cloud registration is to estimate the optimal rigid transformation between two point clouds, P and Q. Let the source point cloud be denoted as , and the target point cloud as . Our task is to seek a rigid transformation such that the transformed source point cloud achieves maximum alignment with the target point cloud Q within the overlapping region.
However, due to their unique acquisition mechanisms and the inherent characteristics of the underwater environment, the nature of MBES point clouds presents two core challenges to this general task: strong anisotropy with sparse structural features, and inherently low-overlap rates.
We define the objective as:
where
denotes a feature-aware registration objective that measures the alignment quality between the transformed source point cloud
and the target point cloud
Q, and
represents the feature descriptors learned by the model. The overlap rate
between two point clouds is defined as:
where
denotes the number of source points,
is the indicator function, and
represents the overlap tolerance threshold. Our method aims to improve registration recall and reduce registration errors under conditions where
is extremely low.
3.2. The Predator Network Architecture
Predator [
15] is a widely used 3D point cloud registration network designed for low-overlap point-cloud pairs, making it a relevant baseline for MBES registration. This architecture learns to detect the overlap region between two unaligned point clouds and focuses its sampling of feature points specifically within this area. The architecture comprises three main components: Encoder, Overlap Attention Module, and Decoder.
The encoder employs a hybrid of two KPConv [
13] blocks—Simple KPConv and ResNet Bottleneck KPConv (RBKPC,
Figure 2a)—to extract features from the two input point clouds, thereby generating their respective superpoints.
The core Overlap Attention Module first uses a Graph Neural Network (GNN,
Figure 2b) [
14] to capture feature neighborhood contexts within each individual point cloud. Subsequently, it employs a multi-head attention mechanism to facilitate information exchange between the superpoints of the two point clouds, enabling the model to determine whether specific superpoints are located within the overlap region. The output point clouds are then processed through another GNN [
14] with distinct parameters to verify whether the neighbors of the superpoints lie within the overlap area, thereby performing a final update on the overlap confidence of the superpoints.
This specific design of the Overlap Attention Module ensures that the model maintains a focused local receptive field during intra-cloud information exchange, while using the multi-head attention module to provide a global receptive field for the critical determination of overlap.
Predator [
15] employs three distinct loss functions to supervise its three respective outputs—
F,
o, and
m. Specifically, a cross-entropy-based loss is applied to both the overlap scores
o and matchability scores
m against their ground-truth values. The core component is the Circle Loss [
41] applied to feature computation. It performs joint optimization on positive and negative pair similarities; through the dynamic adjustment of gradient balancing factors, it ensures the model learns highly discriminative point cloud feature representations.
where
and
represent the weighted exponential aggregation terms for positive and negative pairs, respectively, defined as:
In Equation (
4),
denotes the Euclidean distance between feature vectors in the embedding space. The Circle Loss formulation employs adaptive weighting terms defined as
and
, where
serves as the scaling hyperparameter. Regarding the hyperparameters, we adopt the standard relaxation margins of
and
for positive and negative pairs, respectively. Each of the three loss terms is symmetrized by averaging the query-to-gallery and gallery-to-query directions. Following Predator [
15], the total objective is their equally weighted sum,
.
3.3. A-KPConv
The KPConv [
13] module employs kernel points that are isotropically distributed in space to aggregate neighborhood information. However, when confronted with the non-uniform and distinctly anisotropic distribution of MBES point clouds, the feature representation capability of such isotropic kernels is constrained, potentially leading to the learning of erroneous features. To address this challenge, we design A-KPConv, as illustrated in
Figure 3. The core philosophy of this operator is to enable the shape and orientation of the convolution kernel to adaptively adjust according to the local geometric features of the point cloud.
Let a point in the point cloud be defined as
, with its corresponding feature denoted as
. For any convolution center point
, we define its neighborhood for PCA as:
where
represents the radius of the PCA scope. In this work,
is set equal to the convolution radius of the corresponding KPConv layer, so that the PCA neighborhood coincides exactly with the convolution receptive field and no additional hyperparameter is introduced; a sensitivity analysis of
is provided in
Section 4.6. To eliminate the influence of position and analyze only the shape of the neighborhood, we translate all neighborhood points such that their centroid is located at the origin:
where
is the centroid of the neighborhood. These centroid-based coordinates are used only for shape analysis. Subsequently, we compute the covariance matrix
of the neighborhood point set:
We perform eigendecomposition on to obtain three eigenvectors , representing the first, second, and third principal directions, respectively, along with their corresponding eigenvalues . In the implementation, the eigenvalues are clamped to nonnegative values before the scale computation. The eigenvector signs are canonicalized consistently, and the resulting orthonormal basis is adjusted to satisfy , ensuring that is a proper rotation rather than a reflection. Here, the first principal direction corresponds to the orientation of the strip-like features in MBES data.
Using the PCA results, we construct an affine transformation matrix for each convolution center point . This matrix is composed of a rotation matrix and a scaling matrix . Specifically, serves to align the coordinate system with the principal directions of the local geometry; meanwhile, , where for and is a small positive constant for numerical stability ( in all experiments), is used to perform scaling along the aligned coordinate axes. Finally, the affine transformation matrix for the kernel points is formulated as .
Note that the scale along the first principal direction,
, may become large in nearly degenerate neighborhoods where
. Several properties mitigate this behavior in practice. First, the scaling is self-normalized by the second principal variance, so that neither
nor
exceeds 1, while the growth of
follows the square root of the anisotropy ratio. Second,
prevents singular division, and the nonnegative eigenvalue clamp avoids invalid square-root operations caused by small numerical round-off errors. Finally, the correlation in Equation (
9) is truncated at distance
with
, so an over-stretched kernel point loses its influence on neighboring points rather than producing an unbounded aggregation weight. Training remained numerically stable in all our experiments. Nevertheless, no explicit fallback is used for nearly degenerate neighborhoods; under extreme sparsity or weak local structure, the estimated eigen-directions may still become unstable and the aggregation may lose informative kernel-point responses. We therefore treat such neighborhoods as a limitation rather than claiming unconditional numerical robustness.
In the standard KPConv [
13], the convolution kernel comprises a set of base kernel points
defined in a local coordinate frame anchored at the convolution center. In contrast, within A-KPConv, these base kernel points undergo dynamic deformation via the transformation matrix
, yielding a set of anisotropic kernel points
that adapt to the local geometry and remain anchored at
. Since the eigen-directions and eigenvalues used to construct
are translation-invariant properties of the neighborhood shape, the transformation estimated in the centroid frame can be applied directly in this center-anchored frame. Following the KPConv convention, the influence weights and the feature aggregation are computed in the center-relative coordinates of the neighbor points. The aggregated features can be expressed as:
where
denotes the learnable weight matrix associated with the
m-th kernel point,
is the center-relative position of the neighbor point
, and
denotes the correlation between the kernel point
and
:
where
represents the influence radius of the kernel.
3.4. A-Predator Architecture
Figure 4 illustrates the overall architecture of the improved A-Predator. While our model inherits the general encoder-decoder structure of Predator [
15], the core modifications are concentrated within the encoder section. In the initial stages of the network, where geometric structural information is most explicit, the deployment of the structure-aware A-KPConv is most effective for capturing anisotropic features. Conversely, as the network deepens and representations become increasingly abstract, the standard isotropic KPConv [
13] is sufficient for feature extraction from high-level features, where the benefit of directional modeling diminishes (cf. the replacement-depth ablation in
Section 4.3).
Consequently, we replace the standard KPConv [
13] modules in the first three layers of the Predator encoder with our proposed A-KPConv layers. The first layer applies a Simple A-KPConv block for initial structure-aware feature extraction at the full input resolution. The second and third layers use ResNet Bottleneck A-KPConv (RBAKPC) blocks, which incorporate residual connections around the anisotropic convolution to facilitate gradient flow while preserving the ability to capture directional geometric features; the third block operates with stride and performs the first downsampling in the encoder. Within each RBAKPC block, the internal KPConv modules are replaced with A-KPConv, while the residual pathway and strided convolution for downsampling remain unchanged. Deeper encoder layers retain the original isotropic KPConv.
To ensure a fair evaluation, both the network’s bottleneck layer—specifically the Overlap Attention Module with its GNN-based intra-cloud message passing and cross-attention mechanism—and the subsequent decoder remain identical to those in the original Predator [
15], isolating the effect of the anisotropic convolution replacement so that performance differences can be attributed directly to the structure-aware feature learning in the encoder.