arrow
Return

Layer Normalization for TSK Fuzzy System Optimization in Regression Problems

delete2023-01-01
delete17
PRE
AI
Y
Yuqi Cui
Y
Yifan Xu
R
Ruimin Peng
D
Dongrui Wu *
DOI:10.1109/TFUZZ.2022.3185464delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Recently, mini-batch gradient descent (MBGD)-based optimization has become popular in Takagi-Sugeno-Kang (TSK) fuzzy system optimization. However, it suffers from some challenges, including the curse of dimensionality and the sensitivity to the choice of the optimizer. The former has been alleviated by our previously proposed high-dimensional TSK (HTSK) algorithm. In this article, we point out that the latter is caused by the gradient vanishing problem on the rule consequent parameters, which in turn is caused by the small magnitude of the normalized rule firing levels, especially when the number of rules is large. Thus, the rule consequents are easily trapped into a bad local minimum with poor generalization performance. We propose to use first layer normalization (LN) to amplify the small firing levels, and then rectified linear unit (ReLU) to discard rules far away from the current training sample. We evaluated our proposed HTSK-LN and HTSK-LN-ReLU on twelve regression datasets with various sizes and dimensionalities. Experiments demonstrated that they can significantly improve the generalization performance, regardless of the training set size, feature dimensionality, choice of the optimizer, and rulebase size.
Keywords:
Generalization
high-dimensionality
layer normalization (LN)
optimization
Takagi-Sugeno-Kang (TSK) fuzzy system

Journal

IEEE Transactions on Fuzzy Systems cover
IEEE Transactions on Fuzzy Systems
IF:
11.9
Papers:
4.9K
Citations:
2.9W

Organization

No organization information available