Class NHiTSModel<T>
- Namespace
- AiDotNet.TimeSeries
- Assembly
- AiDotNet.dll
Implements N-HiTS (Neural Hierarchical Interpolation for Time Series) for efficient long-horizon forecasting.
public class NHiTSModel<T> : TimeSeriesModelBase<T>, ITimeSeriesModel<T>, IFullModel<T, Matrix<T>, Vector<T>>, IModel<Matrix<T>, Vector<T>, ModelMetadata<T>>, IModelSerializer, ICheckpointableModel, IParameterizable<T, Matrix<T>, Vector<T>>, IFeatureAware, IFeatureImportance<T>, ICloneable<IFullModel<T, Matrix<T>, Vector<T>>>, IGradientComputable<T, Matrix<T>, Vector<T>>, IJitCompilable<T>
Type Parameters
TThe numeric type used for calculations (e.g., float, double).
- Inheritance
-
NHiTSModel<T>
- Implements
- Inherited Members
- Extension Methods
Remarks
N-HiTS is an evolution of N-BEATS that addresses limitations in long-horizon forecasting through:
- Multi-rate data sampling via hierarchical interpolation
- Stack-specific input pooling to capture patterns at different frequencies
- More efficient parameterization compared to N-BEATS
- Interpolation-based basis functions for smoother predictions
Original paper: Challu et al., "N-HiTS: Neural Hierarchical Interpolation for Time Series Forecasting" (AAAI 2023).
Production-Ready Features:
- Uses Tensor<T> for GPU-accelerated operations via IEngine
- Proper backpropagation via automatic differentiation
- Vectorized operations - no scalar loops in hot paths
- All parameters are trained (not subsets)
For Beginners: N-HiTS improves upon N-BEATS by using a "zoom lens" approach to time series. It looks at your data at three different zoom levels: - Zoomed out (low resolution): Captures long-term trends like yearly seasonality - Medium zoom: Captures medium-term patterns like monthly cycles - Zoomed in (high resolution): Captures short-term fluctuations like daily variations
By combining insights from all three levels, it produces more accurate forecasts, especially for predicting far into the future.
Constructors
NHiTSModel(NHiTSOptions<T>?)
Initializes a new instance of the NHiTSModel class.
public NHiTSModel(NHiTSOptions<T>? options = null)
Parameters
optionsNHiTSOptions<T>Configuration options for N-HiTS.
Properties
ParameterCount
Gets the number of parameters in the model.
public override int ParameterCount { get; }
Property Value
Remarks
This property returns the total count of trainable parameters in the model. It's useful for understanding model complexity and memory requirements.
Methods
CreateInstance()
Creates a new instance of the derived model class.
protected override IFullModel<T, Matrix<T>, Vector<T>> CreateInstance()
Returns
- IFullModel<T, Matrix<T>, Vector<T>>
A new instance of the same model type.
Remarks
This abstract factory method must be implemented by derived classes to create a new instance of their specific type. It's used by Clone and DeepCopy to ensure that the correct derived type is instantiated.
For Beginners: This method creates a new, empty instance of the specific model type. It's used during cloning and deep copying to ensure that the copy is of the same specific type as the original.
For example, if the original model is an ARIMA model, this method would create a new ARIMA model. If it's a TBATS model, it would create a new TBATS model.
DeserializeCore(BinaryReader)
Deserializes model-specific data from the binary reader.
protected override void DeserializeCore(BinaryReader reader)
Parameters
readerBinaryReaderThe binary reader to read from.
Remarks
This abstract method must be implemented by each specific model type to load its unique parameters and state.
For Beginners: This method is responsible for loading the specific details that make each type of time series model unique. It reads exactly what was written by SerializeCore, in the same order, reconstructing the specialized parts of the model.
It's the counterpart to SerializeCore and should read data in exactly the same order and format that it was written.
This separation allows the base class to handle common deserialization tasks while each model type handles its specialized data.
ForecastHorizon(Vector<T>)
Generates forecasts for the full horizon using hierarchical processing.
public Vector<T> ForecastHorizon(Vector<T> input)
Parameters
inputVector<T>
Returns
- Vector<T>
GetModelMetadata()
Gets metadata about the time series model.
public override ModelMetadata<T> GetModelMetadata()
Returns
- ModelMetadata<T>
A ModelMetaData object containing information about the model.
Remarks
This method provides comprehensive metadata about the model, including its type, configuration options, training status, evaluation metrics, and information about which features/lags are most important.
For Beginners: This method provides important information about the model that can help you understand its characteristics and performance.
The metadata includes:
- The type of model (e.g., ARIMA, TBATS, Neural Network)
- Configuration details (e.g., lag order, seasonality period)
- Whether the model has been trained
- Performance metrics from the last evaluation
- Information about which features (time periods) are most influential
This information is useful for documentation, model comparison, and debugging. It's like a complete summary of everything important about the model.
PredictSingle(Vector<T>)
Generates a prediction for a single input vector.
public override T PredictSingle(Vector<T> input)
Parameters
inputVector<T>The input feature vector.
Returns
- T
The predicted value.
Remarks
This abstract method must be implemented by derived classes to generate a prediction for a single input vector using the model-specific algorithm.
For Beginners: This method takes a single row of input data (representing one time point) and calculates what the model predicts will happen at that point. Each type of time series model will have its own way of calculating this prediction based on the patterns it learned during training.
SerializeCore(BinaryWriter)
Serializes model-specific data to the binary writer.
protected override void SerializeCore(BinaryWriter writer)
Parameters
writerBinaryWriterThe binary writer to write to.
Remarks
This abstract method must be implemented by each specific model type to save its unique parameters and state.
For Beginners: This method is responsible for saving the specific details that make each type of time series model unique. Different models have different internal structures and parameters that need to be saved separately from the common elements.
For example:
- An ARIMA model would save its AR, I, and MA coefficients
- A TBATS model would save its level, trend, and seasonal components
- A neural network model would save its weights and biases
This separation allows the base class to handle common serialization tasks while each model type handles its specialized data.
TrainCore(Matrix<T>, Vector<T>)
Trains the N-HiTS model using proper backpropagation via automatic differentiation.
protected override void TrainCore(Matrix<T> x, Vector<T> y)
Parameters
xMatrix<T>yVector<T>