Bayesian Prototypical Pruning for Efficient Video Understanding in Human-Robot Collaboration

Researchers at Northwestern Polytechnical University have proposed a novel end-to-end Bayesian framework, called Bayesian Prototypical Pruning (ProtoPrune), for efficiently pruning tokens in video understanding tasks. This method aims to improve the computation efficiency of video Transformers in human-robot collaborative (HRC) workspaces. By leveraging prototypical contrastive learning and variational dropout, ProtoPrune can achieve a pruning rate of 37.2% while retaining 92.9% of task performance, significantly improving computational efficiency.

Key Takeaways:

  • Bayesian Prototypical Pruning (ProtoPrune) is a novel end-to-end Bayesian framework for token pruning in video understanding tasks.
  • ProtoPrune leverages prototypical contrastive learning and variational dropout to improve robustness and efficiency.
  • The method can achieve a pruning rate of 37.2% while retaining 92.9% of task performance using Uniformer and ActionCLIP.
  • ProtoPrune offers a theoretically grounded and hardware-friendly solution for deploying video Transformers in real-world HRC environments.
  • The research provides convergence analysis to ensure the stability of the method.
  • The authors suggest that additional information can be obtained by contacting Bohua Peng, School of Automation, Northwestern Polytechnical University.

Statistics:

  • 37.2% pruning rate achieved by ProtoPrune.
  • 92.9% task performance retained by ProtoPrune using Uniformer and ActionCLIP.
  • 13.9% improvement in computational efficiency achieved by ProtoPrune.

Sources:

  • Peng, Bohua, et al. "Bayesian Prototypical Pruning for Transformers in Human-Robot Collaboration." Mathematics, vol. 13, no. 9, 2025, p. 1411, doi: 10.3390/math13091411. (Available at https://doi-org(sdpl.idm.oclc.org)/10.3390/math13091411)
  • MDPI AG. Mathematics. (Publisher)