DOI: 10.3390/rs18122044 ISSN: 2072-4292

GS-TreeAttn: Accurate Tree Point Cloud Completion via Structure-Density Coupled Attention

Haozhe Lin, Wenjun Zhang, Weipeng Jing, Linhui Li

Accurate reconstruction of complete tree point clouds is essential for estimating ecosystem structural characteristics from LiDAR data. In urban forestry environments, however, terrestrial laser scanning (TLS) and mobile laser scanning (MLS) frequently produce incomplete observations. Occlusion caused by neighboring trees, together with interference from surrounding urban objects such as buildings and vehicles, often leads to missing regions within scanned point clouds. These defects may further affect the reliability of tree structural analysis and parameter estimation. Although recent learning-based point cloud completion methods have improved reconstruction performance, several limitations remain when they are applied to complex tree structures. Many existing networks depend on farthest point sampling (FPS) for feature extraction, which can result in the loss of fine-scale branching information. Furthermore, local feature aggregation methods based on the traditional k-nearest neighbor (KNN) strategy are highly sensitive to regions with uneven point cloud distribution, such as the canopy region where density variations are significant in tree point clouds. To alleviate these issues, this study proposes GS-TreeAttn, an attention-guided framework specifically for tree point cloud completion. This network models density and structural representation as a coupled problem and employs a structure-guided density-adaptive attention mechanism to jointly capture global structural dependencies and local geometric features. We comprehensively evaluate the proposed method using publicly available datasets and urban forestry data collected under real-world scanning conditions. Experimental results show that even in complex scenarios with severe occlusion and uneven sampling density, GS-TreeAttn generates more complete reconstruction results. This improvement is particularly evident in regions where the canopy and branches mutually occlude each other, where information loss is very common in real-world urban forestry.

More from our Archive