import torch
from models.stochastic_model import StochasticModel
[docs]
class HullWhiteModel(StochasticModel):
"""
Hull-White interest rate model.
This model describes the evolution of interest rates using the following stochastic differential equation:
dr = (θ(t) - ar)dt + σdW
where:
r is the short-term interest rate
θ(t) is a function of time chosen to fit the initial term structure
a is the speed of mean reversion
σ (sigma) is the volatility
W is a Wiener process
"""
[docs]
def __init__(self, a_init=0.1, sigma_init=0.01, r0_init=0.03):
"""
Initialize the Hull-White model.
Args:
a_init (float): Initial value for speed of mean reversion.
sigma_init (float): Initial value for volatility.
r0_init (float): Initial short-term interest rate.
"""
params = {
'a': torch.tensor(a_init, requires_grad=True),
'sigma': torch.tensor(sigma_init, requires_grad=True),
'r0': torch.tensor(r0_init, requires_grad=True)
}
super().__init__(params)
[docs]
def simulate(self, S0, T, N, steps=100):
"""
Simulate interest rate paths using the Hull-White model.
Args:
S0 (float): Initial asset price (not used in this model, included for consistency).
T (float): Time horizon for simulation.
N (int): Number of simulation paths.
steps (int): Number of time steps in each path.
Returns:
torch.Tensor: Simulated interest rates at time T.
"""
dt = T / steps
a = self.params['a']
sigma = self.params['sigma']
r0 = self.params['r0']
dt = torch.tensor(dt, device=self.device)
N = int(N)
steps = int(steps)
# Ensure positive values for mean reversion speed and volatility
a = torch.clamp(a, min=1e-6)
sigma = torch.clamp(sigma, min=1e-6)
# Initialize interest rates
r = torch.zeros(N, steps, device=self.device)
r[:, 0] = r0
# Simulate interest rate paths
for t in range(1, steps):
r_t_minus = r[:, t - 1]
# Note: This implementation assumes θ(t) = 0 for simplicity
dr = -a * r_t_minus * dt + sigma * torch.sqrt(dt) * torch.randn(N, device=self.device)
r[:, t] = r_t_minus + dr
return r[:, -1]
[docs]
def _apply_constraints(self):
"""
Apply constraints to model parameters to ensure they remain in valid ranges.
"""
self.params['a'].data.clamp_(min=1e-6)
self.params['sigma'].data.clamp_(min=1e-6)