O. W. Bhatti, H. M. Torun and M. Swaminathan, "HilbertNet: A Probabilistic Machine Learning Framework for Frequency Response Extrapolation of Electromagnetic Structures," in IEEE Transactions on Electromagnetic Compatibility, vol. 64, no. 2, pp. 405-417, April 2022, doi: 10.1109/TEMC.2021.3119277.
- Wow! this looks like a useful NN for signal integrity applications.
- Introduction has a very through survey of S-parameter extrapolation techniques.
- "Knowledge-based neural networks are used in [11] to provide extrapolated results for the design space parameters like physical length and width of copper traces. While this approach works for design space, it may not be applicable to extrapolation."
- RNN: Recurrent Neural Nets
- Predict the next sample given the previous batches of samples.
- I like this emphasis on providing uncertainty bounds:
- "Historically, neural networks have been used to provide point estimates of their predictions. Such models lack explainability and could generalize incorrectly in the extrapolated space. Hence, it becomes necessary to provide uncertainty estimates around our predictions. This not only provides an insight to the model accuracy outside the training data but also enables the designer to choose whether to simulate more training points.
- This is a nice summary
- "Our contributions include the following.
- 1) Use of specialized long short term memory recurrent neural networks (LSTM-RNN) for frequency response extrapolation
- 2) HT (Hilbert Transform) to correlate real and imaginary part of the signal to enable causal complex-valued extrapolation.
- 3) Harnessing the variational inference-based Bayesian approach to assess the uncertainty of predictions in the extrapolated space.
Section II: Hilbert Transform
- They provide a nice algorithm to compute the discrete Hilbert Transform
Section IV
- Break the response up into shorter sliding sequences and train the RNN on using earlier sequences to predict later sequences.
- I like this because you are training the NN on the data to be extrapolated, not on a boat load of data that could or not be relevant to the particular problem at hand.
- This is the training phase
- Training -> Inference phase
Section IV:B Bayesian Recurrent Network Architecture
- Great explanations. After two read throughs I think I have a good idea of what they are doing. I'd really like to learn more about probabilistic estimation/detection theory
- Is able to extract the variance of the estimation so the uncertainty bound is tracked.
Above is their result from applying this to a PDN network. Not only were they able to capture the resonances but they were able to provide a confidence bound to the prediction! Brilliant. They also provide a link to some Python code.
Overall, this is very impressive. It uses the dataset itself as all of the training data and yields a very compelling result. Their application is to predict high frequency behavior and then let the user decide if they need to simulate to these higher frequencies or not.
My high frequency extrapolation is so I can convert to the time domain for a time response of a given time step. Within my use case I wonder if the insights provided by this paper can be of use to me.