#!/usr/bin/env python
# coding: utf-8
# # **Use ML techniques for layer stackup modeling**
# ## Table of contents:
# ### 1.**[Motivation](#Motivation)** ...
# ### 2.**[Problem Statements](#Problem_Statements)** ...
# ### 3.**[Generate Data](#Generate_Data)** ...
# ### 4.**[Prepare Data](#Prepare_Data)** ...
# ### 5.**[Choose a Model](#Choose_a_Model)** ...
# ### 6.**[Training](#Training)** ...
# ### 7.**[Neural Network](#Neural_Network)** ...
# ### 8.**[Deploy](#Deploy)** ...
# ### 9.**[Conclusion](#Conclusion)** ...
# ## **Motivation:**
# When planning a PCB layer stackup, often time we would like to know the trade-off between various layout options vs their signal integrity performance. For example, a wider trace may provides smaller impedance yet occupate more routing area. Narrow down the spacing between differential pairs may save some spaces but will also increase crosstalk. Most EDA tool involving system level signal integrity analysis provides "transmission line calculator" like shown below for designer to quickly make estimation and determine the trade-off:
# ![TLineCalc](https://github.com/SPISim/ML_LStkModeling/blob/master/assets/images/KiCad_LineCalc.png?raw=true)
#
# However, all such "calculators" I have seen, even in a commercial one, only consider the single trace or one differential pair itself. They do not take take crosstalks into account. More over, stackup parameters such as conductivity and permetivity must be entered individually instead of a range. As a results, user can't not easily visualize relationships between performance parameters vs the stackup properties. Thus, an enhanced version of such "T-Line calculator", which can address the aformentioned gaps will be very useful. Such tool requires a prediction model to link between various stackup parameters to their performance targets. Data science/machine learning techniques can thus be used to build such model.
# ## **Problem Statements:**
# We would like to build a prediction model such that given a set of stackup parameters such as trace width and spacing etc, its performance such as impedance, attenuations, near-end and far-end crosstalk can be quickly estimated. This model can then be deployed into a stand-alone tool for range based sweep such that a visual plot can be generated to provide relations between various parameters to decide design trade-off.
# ## **Generate Data**:
# ### Overview:
# The model to be built here is for nominal (i.e. numerical) prediction with around 10 attributes, i.e. input variables. Various stackup configurations will be generated via sampling and their corresponding stackup model, in the form of frequency dependent R/L/G/C matrices will be simulated via field solver. Such process are deterministics. Post process steps will read these solved model and calculate performance. Here we define performance to be predicted as impedance, attenuation, near-end/far-end crosstalks and propagation speed.
# ### ***Define stakup structure:***
# There are many possible stakup structures as shown below. For more accurate prediction, we are going to generate one prediction model per structure.
# ![Presets](https://github.com/SPISim/ML_LStkModeling/blob/master/assets/images/LStkPresets.png?raw=true)
#
# Use three single-ended traces (victim in the middle) in strip-line setup as an example, various attributes may be defined as shown below:
# ![Setup](https://github.com/SPISim/ML_LStkModeling/blob/master/assets/images/SLSE3Setup.png?raw=true)
#
# These parameters, such as S(Spacing), W(Width), Sigma(Conductivity), Er(Permitivity), H(Height) etc are represented as varaibles to be sampled.
# ### ***Define sampling points:***
# Next step is to define ranges of variable values and sampling points. Since there are about 10 parameters, full combinatorial data will be impractical. Thus we may need to apply sampling algorithms such as design-of-experiments or spacing filling etc to establish best coverage of the solution space.
# For this setup, we have generate 10,000 cases to be simulated.
# ![Sample](https://github.com/SPISim/ML_LStkModeling/blob/master/assets/images/SLSE3Sample.png?raw=true)
# ### ***Generate inputs setup and simulate:***
# Once we have sample points, layer stackup configurations to the solver to be used will be generated. Each field solver has different syntax thus a flow will be needed...
# ![TProFlow](https://github.com/SPISim/ML_LStkModeling/blob/master/assets/images/SLSE3FlowGUI.png?raw=true)
#
# In this case, we use HSpice from Synopsys for field solver, thus each of 10K parameter combinations will be used to generate their spice input files for simulation:
# ![Simulate](https://github.com/SPISim/ML_LStkModeling/blob/master/assets/images/SLSE3Simulate.png?raw=true)
#
# The next step is to perforam circuit simulation for all these cases. This may be a time-consuming process so a distributed environment or simulation farm may be used.
# ### ***Performance measurement:***
# The outcome of each simulation is a frequency dependent tabular model, corresponding to its layer stackup settings. HSpice's tabular format looks like this:
# ![Tabular](https://github.com/SPISim/ML_LStkModeling/blob/master/assets/images/SLSE3Tabular.png?raw=true)
# Next step is to load these models and do performance measurement:
# ![Measure](https://github.com/SPISim/ML_LStkModeling/blob/master/assets/images/SLSE3Measure.png?raw=true)
# Matrix manipulation such as eigen-value decomponsition will be applied in order to obtain the characteristic impedance and propagation speed etc. Measurement output of each model should be a set of parameters which will be combined with original inputs to form the dataset for our prediction modeling.
# ## **Prepare Data:**
# From this point, we can start the modeling process using python and various packages.
# In[101]:
get_ipython().run_line_magic('matplotlib', 'inline')
## Initial set-up for data
import os
import pandas as pd
import matplotlib
import matplotlib.pyplot as plt
import numpy as np
prjHome = 'C:/Temp/WinProj/LStkMdl'
workDir = prjHome + '/wsp/'
srcFile = prjHome + '/dat/SLSE3.csv'
def save_fig(fig_id, tight_layout=True, fig_extension="png", resolution=300):
path = os.path.join(workDir, fig_id + "." + fig_extension)
print("Saving figure", fig_id)
if tight_layout:
plt.tight_layout()
plt.savefig(path, format=fig_extension, dpi=resolution)
# In[102]:
# Let's read the data and do some statistic
srcData = pd.read_csv(srcFile)
# Take a peek:
srcData.head()
# In[103]:
srcData.info()
# In[104]:
srcData.describe()
# ### ** Note that:**
# - Sigma(Conductivity) and H(default layer height) are constants in this setup;
# - FNAME (File name) is not needed for modeling
# - Z0 (impedance) has outliers
# - Forward/Backward crosstalk (Kb/kf) have missing terms
# In[105]:
# drop constant and file name columns
stkData = srcData.drop(columns=['H', 'SIGMA', 'FNAME'])
# In[106]:
# plot distributions before dropping measurement outliers
stkData.hist(bins=50, figsize=(20,15))
save_fig("attribute_histogram_plots")
plt.show()
# In[107]:
# drop outliers and invalid Kb/Kf cells
# These may be caused by unphysical stakup model or calculation during post-processing
maxZVal = 200
minZVal = 10
stkTemp = stkData[(stkData['Z0SE(1_SE)'] < maxZVal) & \
(stkData['Z0SE(1_SE)'] > minZVal) & \
(np.abs(stkData['KBSENB(1_1)']) > 0.0) & \
(np.abs(stkData['KFSENB(1_1)']) > 0.0)]
# Check again to make sure data are now justified
stkTemp.info()
# In[108]:
# now plot distributions again, should see proper distribuition now
stkData = stkTemp
stkData.hist(bins=50, figsize=(20,15))
save_fig("attribute_histogram_plots")
plt.show()
# In[109]:
# find principal components for Z
corr_matrix = stkData.drop(columns=['KBSENB(1_1)', 'KFSENB(1_1)', 'S0SE(1_SE)', 'A(1_1)']).corr()
corr_matrix['Z0SE(1_SE)'].abs().sort_values(ascending=False)
# From this correlation matrix above, it can be shown that trace width and height are dominate factors for the trace's impedance.
# ## **Choose a Model:**
# Since we are building a nominal estimator here, I will try simple linear regressor as estimator first:
# In[110]:
# Separate input and output attributes
allTars = ['Z0SE(1_SE)', 'KBSENB(1_1)', 'KFSENB(1_1)', 'S0SE(1_SE)', 'A(1_1)']
varList = [e for e in list(stkData) if e not in allTars]
varData = stkData[varList]
# In[111]:
# We have 10,000 cases here, try in-memory normal equation directly first:
# LinearRegression Fit Impedance
from sklearn.linear_model import LinearRegression
tarData = stkData['Z0SE(1_SE)']
lin_reg = LinearRegression()
lin_reg.fit(varData, tarData)
# Fit and check predictions using MSE etc
from sklearn.metrics import mean_squared_error, mean_absolute_error
predict = lin_reg.predict(varData)
resRMSE = np.sqrt(mean_squared_error(tarData, predict))
resRMSE
# In[112]:
# Use 10-Split for cross validations:
def display_scores(attribs, scores):
print("Attribute:", attribs)
print("Scores:", scores)
print("Mean:", scores.mean())
print("Standard deviation:", scores.std())
from sklearn.model_selection import cross_val_score
lin_scores = cross_val_score(lin_reg, varData, tarData,
scoring="neg_mean_squared_error", cv=10)
lin_rmse_scores = np.sqrt(-lin_scores)
display_scores(varList, lin_rmse_scores)
# In[113]:
# try Regularization it self
from sklearn.linear_model import Ridge
ridge_reg = Ridge(alpha=1, solver="cholesky")
ridge_reg.fit(varData, tarData)
predict = ridge_reg.predict(varData)
resRMSE = np.sqrt(mean_squared_error(tarData, predict))
resRMSE
# Thus a 3 ohms or so difference may be obtained from this estimator. What if higher order regression is used:
# In[114]:
from sklearn.preprocessing import PolynomialFeatures
poly_features = PolynomialFeatures(degree=2, include_bias=False)
varPoly = poly_features.fit_transform(varData)
lin_reg = LinearRegression()
lin_reg.fit(varPoly, tarData)
predict = lin_reg.predict(varPoly)
resRMSE = np.sqrt(mean_squared_error(tarData, predict))
resRMSE
# A more accurate model thus may be obtained this way.
# ## **Training and Evaluation:**
# In[115]:
from sklearn.metrics import mean_squared_error
from sklearn.model_selection import train_test_split
def plot_learning_curves(model, X, y):
X_train, X_val, y_train, y_val = train_test_split(X, y, test_size=0.2, random_state=10)
train_errors, val_errors = [], []
for m in range(1, len(X_train)):
model.fit(X_train[:m], y_train[:m])
y_train_predict = model.predict(X_train[:m])
y_val_predict = model.predict(X_val)
train_errors.append(mean_squared_error(y_train_predict, y_train[:m]))
val_errors.append(mean_squared_error(y_val_predict, y_val))
plt.plot(np.sqrt(train_errors), "r-+", linewidth=2, label="Training set")
plt.plot(np.sqrt(val_errors), "b-", linewidth=3, label="Validation set")
plt.legend(loc="upper right", fontsize=14)
plt.xlabel("Training set size", fontsize=14)
plt.ylabel("RMSE", fontsize=14)
lin_reg = LinearRegression()
plot_learning_curves(lin_reg, varData, tarData)
plt.axis([0, 8000, 0, 20])
save_fig("underfitting_learning_curves_plot")
plt.show()
# ## **Neural Network:**
# As the difference between prediction to actual measurement is about two ohms, it has met our modeling goals. As an alternative approacy, let's try neural net modeling below
# In[116]:
from keras.models import Sequential
from keras.layers import Dense, Dropout
numInps = len(varList)
nnetMdl = Sequential()
# input layer
nnetMdl.add(Dense(units=64, activation='relu', input_dim=numInps))
# hidden layers
nnetMdl.add(Dropout(0.3, noise_shape=None, seed=None))
nnetMdl.add(Dense(64, activation = "relu"))
nnetMdl.add(Dropout(0.2, noise_shape=None, seed=None))
# output layer
nnetMdl.add(Dense(units=1, activation='sigmoid'))
nnetMdl.compile(loss='mean_squared_error', optimizer='adam')
# Provide some info
#from keras.utils import plot_model
#plot_model(nnetMdl, to_file= workDir + 'model.png')
nnetMdl.summary()
# In[117]:
# Prepare Training (tran) and Validation (test) dataset
varTran, varTest, tarTran, tarTest = train_test_split(varData, tarData, test_size=0.2)
# scale the data
from sklearn import preprocessing
varScal = preprocessing.MinMaxScaler()
varTran = varScal.fit_transform(varTran)
varTest = varScal.transform(varTest)
tarScal = preprocessing.MinMaxScaler()
tarTran = tarScal.fit_transform(tarTran.values.reshape(-1, 1))
# In[118]:
hist = nnetMdl.fit(varTran, tarTran, epochs=50, batch_size=1000, validation_split=0.1)
tarTemp = nnetMdl.predict(varTest, batch_size=1000)
predict = tarScal.inverse_transform(tarTemp)
resRMSE = np.sqrt(mean_squared_error(tarTest, predict))
resRMSE
# In[119]:
plt.plot(hist.history['loss'])
plt.title('Model loss')
plt.ylabel('Loss')
plt.xlabel('Epoch')
plt.legend(['Train', 'Val'], loc='upper right')
plt.show()
# With epoc increased to 100, we can even obtain 1.5 ohms accuracy. It seems this neural network model is comparable to the polynominal regressor and meet our needs.
# In[120]:
# save model and architecture to single file
nnetMdl.save(workDir + "LStkMdl.h5")
# finally
print("Saved model to disk")
# ## **Deploy:**
# Using the SESL3 data set as an example, we follow the similar process and built 10+ prediction models for different stakup structure setup. The polynominal model or neural network can be implemented in Java/C++ to avoid dependencies on python's package for distribution purpose. The implemented front-end, shown below, provide a quick and easy method for system designer for stackup/routing planning:
# ![Deploy](https://github.com/SPISim/ML_LStkModeling/blob/master/assets/images/SLSE3Deploy.png?raw=true)
# ## **Conclusion:**
# In this post/markdown document, we decribe the stackup modeling process using data science/machine learning techniques. The outcome is a deployed front-end with modeled neural network for user's instant performance evaluation. The data set and this markdown document is published on this project's git-hub page.