ADATSK
ADATSK extends TSK with an adaptive softmin antecedent that stabilizes high-dimensional fuzzy inference while preserving first-order TSK consequents.
Reference
G. Xue, Q. Chang, J. Wang, K. Zhang and N. R. Pal, "An Adaptive Neuro-Fuzzy System With Integrated Feature Selection and Rule Extraction for High-Dimensional Classification Problems," in IEEE Transactions on Fuzzy Systems, vol. 31, no. 7, pp. 2167-2181, July 2023, doi: 10.1109/TFUZZ.2022.3220950
Mathematical Formulation
ADATSK extends TSK fuzzy inference by using an adaptive softmin antecedent (Ada-softmin) together with first-order linear consequents.
Antecedent
Each rule-term membership is typically computed with a Gaussian function:
where \(c_{r,d}\) is the center and \(\sigma_{r,d}>0\) is the spread.
In highFIS, the paper-strict ADATSKClassifier path uses a dedicated
ADATSKGaussianMF antecedent, with:
and applies sigma=1 before training, yielding the paper's simplified
form:
Adaptive Ada-softmin aggregation
ADATSK computes rule firing strengths with an adaptive softmin based on the minimum antecedent membership for each rule:
The exponent \(\hat{q}_r\) is recomputed on every forward pass and is clamped for numerical stability, which avoids the fixed-parameter softmin problems of underflow and fake minimum.
Normalization
Rule firing strengths are normalized by simple sum normalization:
Consequent
In the ADATSK paper, the base formulation is presented for classification, with first-order TSK consequents:
Output aggregation
The final prediction is the normalized weighted sum of rule consequents:
For classification:
Code ↔ Paper Correspondence
| Equation | Class / Method | Description |
|---|---|---|
| Adaptive softmin | highfis.layers.AdaSoftminRuleLayer |
Computes per-rule softmin exponents from the minimum membership value |
| Normalization | highfis.defuzzifiers.SumBasedDefuzzifier |
Standard sum-based rule strength normalization |
| Consequent | ClassificationConsequentLayer |
First-order linear consequents for classification |
| Membership functions | highfis.memberships.ADATSKGaussianMF |
Paper-style Gaussian antecedent used by ADATSKClassifier |
Implementation notes
In highFIS, ADATSKClassifierModel implements the paper-aligned ADATSK core
by replacing the standard product antecedent with the adaptive softmin
operator.
Model classes
ADATSKClassifierModeluseshighfis.layers.AdaSoftminRuleLayerto compute rule strengths.- The classifier consequent remains first-order linear and is normalized
with
highfis.defuzzifiers.SumBasedDefuzzifier. ADATSKClassifierModeldoes not expose the feature-selection/ rule-extraction gates of FSRE-ADATSK.
Estimator wrappers
ADATSKClassifieris the paper-strict default wrapper.- Default settings follow the paper protocol:
n_mfs=3,mf_init="grid",rule_base="coco", full-batch updates (the defaultbatch_size="auto"), and no shuffling. - For grid initialization, MF centers are placed directly on
[V_{\min}, V_{\max}]with no margin padding. - In the pre-train hook, Gaussian spreads are set to
sigma=1and frozen. - Consequent parameters are initialized to zero in the ADATSK classifier paper-strict path.
- For high-dimensional inputs (default threshold
1000features), antecedent parameters are frozen by default to match the paper's experimental protocol. - Batch normalization on the consequent inputs is enabled by default
(
consequent_batch_norm=True). This is required for numerical stability: the first-order consequent spans all features and is optimized with plain full-batch gradient descent, which diverges (consequent weights grow unbounded and becomeNaN) on high-dimensional data without normalization, causing the classifier to collapse below the majority-class baseline. Batch normalization is consistent with the CoCo-FRB TSK classifier lineage the paper builds on (Cui et al., 2020, ref. [10]). It can be disabled for low-dimensional problems where divergence does not occur.
Membership functions
- The primary antecedent MFs are
highfis.memberships.ADATSKGaussianMFobjects in the paper-strict classifier path. - Gaussian spreads are fixed at
sigma=1in the paper-strict ADATSK path, reproducing Eq. (3).
Training in the paper vs. highFIS
- The paper trains ADATSK end-to-end using full-batch gradient descent and MSE-style classification error.
- The paper-strict ADATSK default in highFIS uses
HalfSumSquaredErrorLossfor the classifier and SGD-based full-batch optimization. -
epsis used to clamp membership values and stabilize log-space computations inAdaSoftminRuleLayer. -
FSRE-ADATSK is documented separately in
docs/models/fsre-adatsk.md.
Framework extensions (outside paper-strict scope)
ADATSKRegressorModelandADATSKRegressorare provided as framework extensions for regression workflows.highfis.memberships.CompositeGaussianMFremains available as an engineering alternative for custom experiments.
Alignment with the paper
- The paper's key ADATSK contribution is the adaptive softmin antecedent operator to avoid numeric underflow and fake minimum effects.
- highFIS implements this via
AdaSoftminRuleLayerwith a per-rule exponent derived from the rule's minimum antecedent membership. - The default
ADATSKClassifiernow follows the paper protocol (CoCo rule base, Eq. (3)-style Gaussian antecedent withsigma=1fixed, full-batch GD, MSE-style classification loss, and zero-initialized consequents), plus consequent batch normalization for the numerical stability needed to reproduce the paper's high-dimensional accuracies (e.g., Colon in Table III). - Regression and alternative MF variants are treated as explicit framework
extensions, not part of the strict ADATSK paper baseline.
ADATSKRegressoris built like the classifier: the same Gaussian antecedents and zero-initialized consequents, with the bias of every rule set to the mean of the target before training.
Fidelity to the source article
The formulas of the package equal those of the article: the membership function \(e^{-(x-m)^2}\), the Ada-softmin of Eqs. (15) and (16) within \(10^{-8}\), the centres evenly placed over the range of each feature, consequents starting at zero and plain gradient descent on the whole training set. With a learning rate of 0.05 and 1000 iterations, which the article does not state, the published accuracies of Iris and Wdbc are reproduced within one point; see the reproduction.
The spreads stay fixed at one during training, since the membership function of the
article has none. The classifier uses highfis.losses.HalfSumSquaredErrorLoss, Eq. (8)
of the article; with plain gradient descent the scale of the loss sets the size of the
step.
The defaults of the estimator are not the article's, which does not state its learning
rate: the step follows the number of features (learning_rate="auto",
min(0.1, 1 / (n_features + 1))), which cannot diverge, there are 300 epochs, and the
consequent inputs are normalized. A fixed step is either too small for a few features
or diverges on several thousand.