Weave Mamba Fusion: Global Cross-Scale Interaction for Lightweight Face Detection
Organizations: School of Data Science, The University of Suwon, Korea
Abstract
Feature pyramid methods, from FPN to BiFPN, have achieved strong performance in face detection by fusing multi-scale features. However, detecting faces under unconstrained conditions, such as small scale, occlusion, and extreme pose, remains difficult, as it requires global cross-scale dependencies that local fusion cannot model. State space models such as Mamba provide global context with linear complexity by scanning features as a sequence, and therefore offer a promising direction for this problem. Nevertheless, such a scan needs the two pyramid scales combined into a single feature map, and the way they are combined determines whether cross-scale structure is preserved. Summation collapses the two scales before the scan, so the scan has no cross-scale structure to exploit, while concatenation keeps both scales but at far higher cost. To address this, we propose \textbf{Weave Mamba Fusion (WMF)}, which interleaves two adjacent pyramid scales column by column so that each step of a horizontal bidirectional SS2D scan moves from one scale to the other. With partial-channel processing and parameter-free de-weaving, WMF enables efficient cross-scale interaction while preserving feature structure. Integrating WMF into every fusion node yields \textbf{WeaveBiFPN}, the neck of our \textbf{WeaveFace} detector. On WIDER FACE, WeaveFace achieves 91.41% mean AP with only 0.34M parameters and 1.16 GFLOPs, outperforming prior detectors under 0.5M parameters. Its largest gains are on the Hard subset, where it reaches 87.14% AP. The code is publicly available at https://github.com/dohun-mat/WeaveMambaFusion.
Figures & tables
| Model | Backbone | Neck | Fusion | Easy | Medium | Hard | Avg. (%) | #Params (M) | GFLOPs |
| DSFD ‡ [ 13 ] | ResNet-152 | FEM | Conv. | 96.60 | 95.70 | 90.40 | 94.23 | 120.06 | 259.55 |
| TinaFace ‡ [ 37 ] | ResNet-50 | FPN | Conv. | 97.00 | 96.30 | 93.40 | 95.57 | 37.98 | 172.95 |
| TransEnc-R50 † | ResNet-50 | FPN | Self-attention | 93.03 | 92.89 | 88.56 | 91.49 | 33.72 | 32.92 |
| RetinaFace ‡ [ 6 ] | ResNet-50 | FPN | Conv. | 96.70 | 96.10 | 91.40 | 94.73 | 29.50 | 37.59 |
| SCRFD-10GF [ 9 ] | Basic Res | PANet | Conv. | 95.93 | 94.95 | 90.81 | 93.90 | 3.86 | 9.98 |
| FaceBoxes ‡ [ 34 ] | - | - | - | 85.90 | 81.60 | 55.70 | 74.40 | 1.01 | 0.28 |
| Method | AFW AP (%) | PASCAL AP (%) | #Params |
| MogFace [ 17 ] | 99.85 | 99.32 | 85.26M |
| EfficientSRFace-L [ 24 ] | 99.94 | 98.84 | 18.84M |
| SCRFD-0.5GF [ 9 ] | 98.60 | 98.54 | 0.57M |
| SCRFD-1.0GF [ 9 ] | 99.70 | 98.60 | 0.64M |
| SCRFD-2.5GF [ 9 ] | 99.82 | 98.91 | 0.67M |
| FaceBoxes [ 34 ] | 98.91 | 96.30 | 1.01M |
| Fusion | Easy | Medium | Hard | Avg. | #Params (M) | FLOPs (G) | Lat. (ms) |
| Sum (w/ sep. conv.) | 92.88 | 90.87 | 84.76 | 89.50 | 0.217 | 0.564 | 15.1 |
| Cross-attention | 93.47 | 91.55 | 86.46 | 90.49 | 0.577 | 11.228 | 23.7 |
| WMF | 94.33 | 92.76 | 87.14 | 91.41 | 0.344 | 1.159 | 23.3 |
| Topology | Fusion | Easy | Medium | Hard | Avg. | #Params (M) | FLOPs (G) |
| FPN | Sum | 91.17 | 89.31 | 83.81 | 88.10 | 0.229 | 0.672 |
| WMF | 91.47 | 89.72 | 84.12 | 88.44 | 0.187 | 0.639 | |
| PANet | Concat | 92.33 | 90.35 | 84.88 | 89.19 | 0.562 | 1.071 |
| WMF | 92.75 | 91.08 | 85.03 | 89.62 | 0.329 | 0.751 | |
| BiFPN | Sum | 92.88 | 90.87 | 84.76 | 89.50 | 0.217 | 0.564 |
| WMF | 94.33 | 92.76 | 87.14 | 91.41 | 0.344 | 1.159 |
| Fusion Operator | Easy | Medium | Hard | Avg. | #Params (M) | FLOPs (G) | |
| Sum + SS2D | 1 | 93.27 | 91.54 | 85.40 | 90.07 | 0.409 | 1.152 |
| 2 | 91.71 | 89.82 | 83.85 | 88.46 | 0.245 | 0.711 | |
| 4 | 90.18 | 88.13 | 81.20 | 86.50 | 0.193 | 0.566 | |
| Concat + SS2D | 1 | 95.14 | 93.79 | 88.40 | 92.44 | 1.080 | 2.649 |
| 2 | 93.49 | 91.57 | 85.81 | 90.29 | 0.510 | 1.350 | |
| 4 | 93.22 | 91.00 | 84.47 | 89.56 | 0.344 | 0.925 |
| Easy | Medium | Hard | Avg. | #Params (M) | FLOPs (G) | |
| 1 | 93.87 | 92.07 | 86.51 | 90.82 | 0.323 | 0.991 |
| 2 | 94.33 | 92.76 | 87.14 | 91.41 | 0.344 | 1.159 |
| 4 | 94.01 | 92.52 | 87.06 | 91.20 | 0.386 | 1.496 |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Backbone | Easy | Medium | Hard | Avg. | #Params (M) | FLOPs (G) |
| EResNet | 94.33 | 92.76 | 87.14 | 91.41 | 0.344 | 1.159 |
| MobileNetV1-0.25 | 95.06 | 93.64 | 88.52 | 92.41 | 0.566 | 1.231 |
| ResNet-50 | 96.29 | 95.44 | 91.20 | 94.31 | 29.756 | 36.275 |
| ResNet-152 | 96.33 | 95.44 | 91.15 | 94.31 | 64.392 | 81.871 |
| Level | Stride | Feature map | Anchor sizes (px) |
| 8 | 16, 32 | ||
| 16 | 64, 128 | ||
| 32 | 256, 512 |
| COCO val2017 | PASCAL VOC | |||||||
| Variant | AP | AP 50 | AP 75 | Params (M) | FLOPs (G) | mAP | Params (M) | FLOPs (G) |
| EfficientDet (baseline) | 34.5 | 52.9 | 36.6 | 3.88 | 2.57 | 76.90 | 3.84 | 2.35 |
| EfficientDet + WMF | 36.2 | 53.6 | 38.2 | 4.25 | 3.21 | 78.63 | 4.21 | 2.99 |