List of usage examples for org.apache.commons.math3.optim.nonlinear.scalar.noderiv CustomPowellOptimizer CustomPowellOptimizer
public CustomPowellOptimizer(double rel, double abs, double lineRel, double lineAbs)
From source file:gdsc.smlm.ij.plugins.EMGainAnalysis.java
/** * Fit the EM-gain distribution (Gaussian * Gamma) * /*from w w w .j a v a 2s.c o m*/ * @param h * The distribution */ private void fit(int[] h) { final int[] limits = limits(h); final double[] x = getX(limits); final double[] y = getY(h, limits); Plot2 plot = new Plot2(TITLE, "ADU", "Frequency"); double yMax = Maths.max(y); plot.setLimits(limits[0], limits[1], 0, yMax); plot.setColor(Color.black); plot.addPoints(x, y, Plot2.DOT); Utils.display(TITLE, plot); // Estimate remaining parameters. // Assuming a gamma_distribution(shape,scale) then mean = shape * scale // scale = gain // shape = Photons = mean / gain double mean = getMean(h) - bias; // Note: if the bias is too high then the mean will be negative. Just move the bias. while (mean < 0) { bias -= 1; mean += 1; } double photons = mean / gain; if (simulate) Utils.log("Simulated bias=%d, gain=%s, noise=%s, photons=%s", (int) _bias, Utils.rounded(_gain), Utils.rounded(_noise), Utils.rounded(_photons)); Utils.log("Estimate bias=%d, gain=%s, noise=%s, photons=%s", (int) bias, Utils.rounded(gain), Utils.rounded(noise), Utils.rounded(photons)); final int max = (int) x[x.length - 1]; double[] g = pdf(max, photons, gain, noise, (int) bias); plot.setColor(Color.blue); plot.addPoints(x, g, Plot2.LINE); Utils.display(TITLE, plot); // Perform a fit CustomPowellOptimizer o = new CustomPowellOptimizer(1e-6, 1e-16, 1e-6, 1e-16); double[] startPoint = new double[] { photons, gain, noise, bias }; final int maxEval = 3000; // Set bounds double[] lower = new double[] { 0, 0.5 * gain, 0, bias - noise }; double[] upper = new double[] { (limits[1] - limits[0]) / gain, 2 * gain, gain, bias + noise }; // Restart until converged. // TODO - Maybe fix this with a better optimiser. This needs to be tested on real data. PointValuePair solution = null; for (int iter = 0; iter < 3; iter++) { IJ.showStatus("Fitting histogram ... Iteration " + iter); try { // Basic Powell optimiser MultivariateFunction fun = getFunction(limits, y, max, maxEval); PointValuePair optimum = o.optimize(new MaxEval(maxEval), new ObjectiveFunction(fun), GoalType.MINIMIZE, new InitialGuess((solution == null) ? startPoint : solution.getPointRef())); if (solution == null || optimum.getValue() < solution.getValue()) { solution = optimum; } } catch (Exception e) { } try { // Bounded Powell optimiser MultivariateFunction fun = getFunction(limits, y, max, maxEval); MultivariateFunctionMappingAdapter adapter = new MultivariateFunctionMappingAdapter(fun, lower, upper); PointValuePair optimum = o.optimize(new MaxEval(maxEval), new ObjectiveFunction(adapter), GoalType.MINIMIZE, new InitialGuess(adapter .boundedToUnbounded((solution == null) ? startPoint : solution.getPointRef()))); double[] point = adapter.unboundedToBounded(optimum.getPointRef()); optimum = new PointValuePair(point, optimum.getValue()); if (solution == null || optimum.getValue() < solution.getValue()) { solution = optimum; } } catch (Exception e) { } } IJ.showStatus(""); IJ.showProgress(1); if (solution == null) { Utils.log("Failed to fit the distribution"); return; } double[] point = solution.getPointRef(); photons = point[0]; gain = point[1]; noise = point[2]; bias = (int) Math.round(point[3]); String label = String.format("Fitted bias=%d, gain=%s, noise=%s, photons=%s", (int) bias, Utils.rounded(gain), Utils.rounded(noise), Utils.rounded(photons)); Utils.log(label); if (simulate) { Utils.log("Relative Error bias=%s, gain=%s, noise=%s, photons=%s", Utils.rounded(relativeError(bias, _bias)), Utils.rounded(relativeError(gain, _gain)), Utils.rounded(relativeError(noise, _noise)), Utils.rounded(relativeError(photons, _photons))); } // Show the PoissonGammaGaussian approximation double[] f = null; if (showApproximation) { f = new double[x.length]; PoissonGammaGaussianFunction fun = new PoissonGammaGaussianFunction(1.0 / gain, noise); final double expected = photons * gain; for (int i = 0; i < f.length; i++) { f[i] = fun.likelihood(x[i] - bias, expected); //System.out.printf("x=%d, g=%f, f=%f, error=%f\n", (int) x[i], g[i], f[i], // gdsc.smlm.fitting.utils.DoubleEquality.relativeError(g[i], f[i])); } yMax = Maths.maxDefault(yMax, f); } // Replot g = pdf(max, photons, gain, noise, (int) bias); plot = new Plot2(TITLE, "ADU", "Frequency"); plot.setLimits(limits[0], limits[1], 0, yMax * 1.05); plot.setColor(Color.black); plot.addPoints(x, y, Plot2.DOT); plot.setColor(Color.red); plot.addPoints(x, g, Plot2.LINE); plot.addLabel(0, 0, label); if (showApproximation) { plot.setColor(Color.blue); plot.addPoints(x, f, Plot2.LINE); } Utils.display(TITLE, plot); }