LBFormer: Scene Perception Segmentation Transformer Based on Local Block
Scene perception for autonomous vehicles and vessels is crucial for autonomous navigation. Current mainstream transformer methods typically split the feature map into windows, such as local, dilated, and horizontal/vertical bar windows. However, their token interaction is confined to fixed windows, posing challenges for image-based semantic segmentation. This article proposes a novel model, LBFormer, which enables flexible token interaction across different windows. Specifically, a window-level affinity graph is constructed from coarse-grained features using self-attention clustering and evolves during training, retaining top-k windows with high semantic relevance for each window. Self-attention purification is then employed to compress and filter fine-grained features with low semantic relevance within the top-k windows, ensuring effective token interaction for each feature point. To enhance context modeling within windows and build a more effective window-level affinity graph, a dual branch method extracts multidimensional features from each window, which are then interacted with and fused via the feature aggregation module. Extensive experiments at an image resolution of 224×224 were conducted on our private YZ-DATA water surface scene dataset and the public CamVid urban scene dataset. The results show that LBFormer achieves an MIoU of 89.80% on YZ-DATA and 61.31% on CamVid, surpassing mainstream transformer methods.
Authors
- Yaonan Wang (ORCID: https://orcid.org/0000-0002-0519-6458)
- Hongjin Wang (ORCID: https://orcid.org/0000-0002-8071-1536)
- Shi Wang (ORCID: https://orcid.org/0000-0002-5117-6095)
- Baoyuan Deng (ORCID: https://orcid.org/0000-0002-7446-8787)
- Yunze He (ORCID: https://orcid.org/0000-0002-7081-8225)
- Xiaofei Zhang (ORCID: https://orcid.org/0000-0002-8256-0401)
- Liang Cheng (ORCID: https://orcid.org/0000-0003-0089-3171)
- Yongjie Zhang (ORCID: https://orcid.org/0009-0005-8949-6878)
Institutions
- Hunan University (CN)
- Shanghai Ocean University (CN)
Publication Details
- Journal
- IEEE Transactions on Industrial Informatics
- Published
- 2024-10-25
- DOI
- https://doi.org/10.1109/tii.2024.3477554
- Citations
- 7
- Primary Topic
- Industrial Vision Systems and Defect Detection
- Type
- article
- Field-Weighted Citation Impact
- 2.44
Funders
- National Natural Science Foundation of China
- Natural Science Foundation of Hunan Province
- Basic and Applied Basic Research Foundation of Guangdong Province