Safe and accelerated screening framework for support tensor machines.

Journal: Neural networks : the official journal of the International Neural Network Society
Published Date:

Abstract

Support Tensor Machines (STMs) constitute an effective supervised learning method for classifying high-dimensional tensor data. However, traditional iterative solving methods are often time-consuming. To effectively address the issue of lengthy training times, inspired by the safe screening strategies employed in support vector machines, we generalize the safe screening rule to the tensor domain and propose a novel safe screening rule for STM, which includes the dual static screening rule (DSSR), the dynamic screening rule (DGSR), and a subsequent checking verification. The screening rule initially employs variational inequalities to screen out a portion of redundant samples before training, reducing the problem scale. During the training process, the rule further accelerates training by iteratively screening redundant samples using the duality gap. We also design a subsequent checking technique based on optimality conditions to guarantee the safety of the screening rule. Building on this, we also develop a flexible safe screening framework, referred to as DS-DGSR, which incorporates the DSSR and the DGSR. It not only tackles the challenges of combining various tensor decomposition methods and the diverse scenarios of the decomposed coefficient parameter and decomposed samples in STMs, but also offers flexible adaptation and application according to the characteristics of different STMs. Numerical experiments on multiple real-world high-dimensional tensor datasets confirm the effectiveness and feasibility of DS-DGSR.

Authors

  • Xiao Li
    Department of Inner Mongolia Clinical Medicine College, Inner Mongolia Medical University, Hohhot, Inner Mongolia, China.
  • Hongmei Wang
  • Yitian Xu
    College of Science, China Agricultural University, Beijing 100083, China. Electronic address: xytshuxue@126.com.