SONATA-based LiDAR Descriptors in SLAM

• SONATA-based LiDAR descriptors are transformer-derived, geometry-aware embeddings designed for fine-grained 3D spatial correspondence in SLAM.
• They leverage local attention and voxel aggregation to capture geometric context while ensuring invariance to density and viewpoint changes.
• Integration within the MPRF pipeline significantly reduces angular errors and runtime compared to classical handcrafted descriptors.

SONATA-based LiDAR descriptors are high-dimensional geometric feature representations generated by a transformer-based backbone (SONATA: Self-supervised point cloud trANsformer), designed for interpretable, fine-grained 3D spatial correspondence in Simultaneous Localization and Mapping (SLAM) tasks. In the MPRF pipeline for multimodal loop closure detection, SONATA descriptors deliver robust geometric verification of candidate loop closures in severely unstructured or texture-deficient environments, leveraging local geometric context learned in a fully self-supervised way.

1. Motivation and Context

The deployment of SLAM systems in GNSS-denied environments, such as planetary-like terrains, is challenged by aliasing and weak textures in visual data, and sparsity or ambiguity in LiDAR geometry. Classical handcrafted descriptors (e.g., FPFH) for 3D point clouds are susceptible to density and viewpoint changes, resulting in degraded performance on scene disambiguation and pose estimation. While learned global 3D features such as PointNetVLAD and MinkLoc have improved retrieval accuracy, they fail to capture the fine-grained geometric cues vital for robust 6-DoF verification. SONATA-based descriptors were introduced to overcome these limitations by encoding each local neighborhood in the LiDAR point cloud into a geometry-aware, high-dimensional embedding that is largely invariant to point density and viewpoint, utilizing a masked autoencoding transformer objective. These descriptors act as the basis for spatial correspondences essential in geometric verification post visual candidate retrieval in the MPRF pipeline (Gonzalez et al., 7 Nov 2025).

2. Mathematical Underpinnings

Let $P = \{p_i \in \mathbb{R}^3\}, i=1 \ldots N$ represent the set of LiDAR points. SONATA-based descriptors are computed in two stages: neighborhood attention and voxel-wise aggregation.

Initial Embedding

Each point $p_i$ is mapped to an initial hidden representation $h_i^{(0)}$ by a multilayer perceptron (MLP):

$$h_i^{(0)} = \phi_{in}(p_i), \quad \phi_{in} : \mathbb{R}^3 \rightarrow \mathbb{R}^{D^l}$$

where typically $D^l = 64$.

Point-Transformer Layers

For each layer $\ell = 0 \ldots L-1$:

• k-nearest neighbors $N(i)$ are found.
• Query, key, value projections are computed:

$$q_i = W_q h_i^{(\ell)}, \quad K_j = W_k h_j^{(\ell)}, \quad V_j = W_v h_j^{(\ell)}$$

• Attention weights:

$$\alpha_{ij} = \mathrm{softmax}_{j \in N(i)} \left( \frac{q_i^\top K_j}{\sqrt{d}} \right)$$

• Neighborhood aggregation:

$m_i = \sum_{j \in N(i)} \alpha_{ij} V_j$

• Feed-forward update:

$p_i$0

Voxel Aggregation

Points are binned into voxels of size $p_i$1 (e.g., $p_i$2). For each voxel $p_i$3 with point set $p_i$4, the voxel feature is averaged:

$p_i$5

where $p_i$6. Each descriptor is finally $p_i$7-normalized:

$p_i$8

3. Pipeline Implementation Details

SONATA-based LiDAR descriptor computation within MPRF consists of the following algorithmic sequence:

1. Downsample or voxelize the raw LiDAR scan into voxels of side $p_i$9.
2. Encode each point using the initial MLP.
3. For each transformer layer, update point embeddings through local attention over $h_i^{(0)}$0 neighbors.
4. For each voxel, average the embeddings of its contained points and normalize the descriptor.
5. The resulting set of $h_i^{(0)}$1-normalized voxel descriptors $h_i^{(0)}$2 are used for subsequent matching.

Key parameters (default values inherited from SONATA):

Parameter	Value	Effect
voxel size $h_i^{(0)}$3	0.20 m	Descriptor density
neighbor radius $h_i^{(0)}$4	0.5 m	Receptive field size
max neighbors $h_i^{(0)}$5	32	Context richness / cost
transformer layers $h_i^{(0)}$6	6	Depth / latency trade-off
attention heads	8	Multi-viewpoint context
$h_i^{(0)}$7 output	64	Initial embedding size
Descriptor dim	512	Matching capacity
$h_i^{(0)}$8 normalization	Yes	Cosine matching basis

Increasing $h_i^{(0)}$9 results in fewer descriptors but coarser alignment. Higher $$h_i^{(0)} = \phi_{in}(p_i), \quad \phi_{in} : \mathbb{R}^3 \rightarrow \mathbb{R}^{D^l}$$0 or $$h_i^{(0)} = \phi_{in}(p_i), \quad \phi_{in} : \mathbb{R}^3 \rightarrow \mathbb{R}^{D^l}$$1 yields greater geometric robustness but increased computational cost. Deeper networks ($$h_i^{(0)} = \phi_{in}(p_i), \quad \phi_{in} : \mathbb{R}^3 \rightarrow \mathbb{R}^{D^l}$$2) provide more global context but risk potential oversmoothing of local features.

4. Role in Geometric Verification

SONATA descriptors are deployed exclusively for geometric verification after visual retrieval in MPRF, based on the following methodology:

• Project 2D image patch locations (from visual retrieval) into the 3D LiDAR scan via calibration, yielding image-backed 3D points for query and candidate scans.
• For each scan, extract the SONATA-based voxel descriptors and normalize.
• For each query descriptor, identify its best candidate match via cosine similarity, applying one-to-one Hungarian assignment and filtering by threshold $$h_i^{(0)} = \phi_{in}(p_i), \quad \phi_{in} : \mathbb{R}^3 \rightarrow \mathbb{R}^{D^l}$$3.
• Establish corresponding 3D point pairs and perform robust transformation estimation using RANSAC with a 0.05 m inlier threshold (minimum 3 points).
• Optionally apply ICP for pose refinement.

This framework enables explicit 3D–3D correspondences, allowing interpretable integration into SLAM pose-graph backends.

5. Evaluation and Comparative Performance

Empirical evaluation (see Table III in (Gonzalez et al., 7 Nov 2025)) on the S3LI test set compares SONATA-based verification to classic and multimodal approaches:

Model	Yaw Error (°)	dx Error (m)	dy Error (m)	#Poses	Time (ms)
FPFH+RANSAC (handcrafted)	46.82	8.23	14.27	100%	12,233
SONATA only + RANSAC	16.36	8.86	15.30	100%	3,572
DINO–LiDAR projection + RANSAC	25.10	8.40	14.27	100%	5,687
MPRF (DINO+SONATA fusion)	8.20	8.44	14.24	100%	3,114

SONATA alone reduces angular (yaw) error by over a factor of two compared to FPFH, and runtime by 3.4x, while DINO–LiDAR projection fails to exploit geometric cues adequately. Fusing appearance (DINO) and geometry (SONATA) further halves the yaw error to 8.2°, demonstrating that interpretable, geometry-grounded SLAM loop closure is attainable at practical computational cost.

Ablation analysis in Sec IV-F confirms that SONATA-only achieves ~16° yaw error; multimodal fusion is critical for state-of-the-art angular accuracy.

6. Computational Requirements and Latency

For typical scan sizes ($$h_i^{(0)} = \phi_{in}(p_i), \quad \phi_{in} : \mathbb{R}^3 \rightarrow \mathbb{R}^{D^l}$$4 points, $$h_i^{(0)} = \phi_{in}(p_i), \quad \phi_{in} : \mathbb{R}^3 \rightarrow \mathbb{R}^{D^l}$$5 voxels), with $$h_i^{(0)} = \phi_{in}(p_i), \quad \phi_{in} : \mathbb{R}^3 \rightarrow \mathbb{R}^{D^l}$$6 neighbors and $$h_i^{(0)} = \phi_{in}(p_i), \quad \phi_{in} : \mathbb{R}^3 \rightarrow \mathbb{R}^{D^l}$$7 transformer layers, the SONATA-based pipeline yields (measured on an RTX A4000):

• kNN search: $$h_i^{(0)} = \phi_{in}(p_i), \quad \phi_{in} : \mathbb{R}^3 \rightarrow \mathbb{R}^{D^l}$$8, ~100 ms
• Transformer updates: $$h_i^{(0)} = \phi_{in}(p_i), \quad \phi_{in} : \mathbb{R}^3 \rightarrow \mathbb{R}^{D^l}$$9, ~250 ms
• Voxel aggregation + normalization: ~50 ms
• Total SONATA extraction: 400–600 ms per scan

Matching (Hungarian algorithm on $D^l = 64$05000 query-candidate pairs pruned by thresholding) consumes ~1.2 s, and RANSAC + ICP ~0.5 s; the full LiDAR-only geometric verification is ~3.6 s per query. MPRF’s two-stage (visual screening + SONATA verification) retrieval is slightly faster at 3.1 s per query, aligning with offline loop closure requirements in planetary SLAM scenarios.

7. Impact, Limitations, and Application Scope

SONATA-based LiDAR descriptors, as integrated in MPRF, are pivotal for fine-grained, correspondence-based 6-DoF pose verification in environments where visual recognition is unreliable. They are not used for global retrieval due to inadequate discriminative capacity and runtime cost at scale. Instead, their principal value lies in geometric verification, enabling interpretable SLAM pose estimation with reduced angular error and competitive computational performance. Empirical gains over FPFH demonstrate superior robustness to viewpoint and density variations. However, this approach is most suited to offline or batch-mode loop closure scenarios rather than fully real-time systems, given current runtimes.

A plausible implication is that further optimizations in transformer inference, voxelization strategies, or context-aware fusion could improve SONATA’s applicability to broader SLAM regimes or real-time operation. Limitations include high memory and runtime demand for dense local matching, constraining scalability in extremely large-scale or high-frequency SLAM deployments. Nevertheless, SONATA-based descriptors represent a substantial advancement in leveraging self-supervised, transformer-derived geometric features for robust SLAM in challenging unstructured domains.