Machine Learning for Fiber Mechanics: From Data Preparation to Model Deployment

The Problem

Fiber biomaterial mechanics (Young's modulus, ultimate strength, toughness) depend on many factors: fiber diameter, orientation, crosslink density, water content, strain rate. Traditional regression handles linear relationships poorly, and fiber networks exhibit highly nonlinear behavior.

The special challenge in materials ML: data is scarce. You might have only 30-50 experimental samples.

import pandas as pd
import numpy as np
from sklearn.ensemble import RandomForestRegressor, GradientBoostingRegressor
from sklearn.gaussian_process import GaussianProcessRegressor
from sklearn.model_selection import cross_val_score

Data Preparation

data = pd.DataFrame({
    "fiber_diameter_nm": [45, 52, 48, 55, 50],
    "orientation_std_deg": [12, 22, 8, 18, 15],
    "crosslink_density_pct": [2.5, 1.8, 3.2, 2.0, 4.1],
    "youngs_modulus_GPa": [2.8, 1.9, 3.5, 2.3, 4.0],
})

Model Selection for Small Data

Model	Small-data advantage	Watch out for
------	---------------------	--------------
Random Forest	Insensitive to hyperparams, feature importance	Keep max_depth small
Gradient Boosting	High accuracy, handles nonlinearity	Easy to overfit, low lr needed
Gaussian Process	Built-in uncertainty, perfect for n<100	O(n^3), avoid n>1000

models = {
    "RF": RandomForestRegressor(n_estimators=200, max_depth=5, random_state=42),
    "GB": GradientBoostingRegressor(n_estimators=100, max_depth=3, learning_rate=0.05, random_state=42),
}
for name, model in models.items():
    scores = cross_val_score(model, X_train, y_train, cv=5, scoring="neg_mean_absolute_error")
    print(f"{name}: MAE = {-scores.mean():.3f} +/- {scores.std():.3f}")

Uncertainty Quantification

def predict_with_uncertainty(model, X, n_iter=100):
    predictions = np.zeros((n_iter, len(X)))
    for i in range(n_iter):
        idx = np.random.choice(len(X_train), len(X_train))
        model.fit(X_train[idx], y_train[idx])
        predictions[i] = model.predict(X)
    return np.mean(predictions, axis=0), np.std(predictions, axis=0)

Advanced Directions

• GNN: model fiber networks as graphs
• Transfer learning: pretrain on simulations, fine-tune on experiments
• Active learning: let the model suggest the next experiment
• PINNs: inject constitutive equations as NN constraints

Key Takeaways

1. Small data is not a blocker -- Bayesian methods and GPs are designed for it
2. Feature engineering matters more than model choice
3. Always report uncertainty -- reviewers will ask
4. Open-source your code and data for reproducibility

• Yang Y., Bai R., Gao W. et al. (2024). Advanced Science, 2413293.