Package: ci_rvm
Contents
Package: ci_rvm#
Installation#
The package can be installed via pip. To install the package, you can use
pip install ci_rvm
Usage#
The most convenient way to use the algorithm is to use the method find_CI. An example is below.
# Example for finding profile likelihood confidence intervals for a
# negative binomial model
# We import some packages for convenience
import numpy as np # for numerical operations
from scipy import stats # for stats functions
from scipy import optimize as op # to maximize the likelihood
from ci_rvm import find_CI # to determine confidence intervals
from ci_rvm import find_function_CI # to determine confidence intervals
# for functions of parameters
# Define the size of the data set
n = 100
# Define the true parameters
k, p = 5, 0.1
# Generate the data set
data = np.random.negative_binomial(k, p, size=n)
# Because the parameters are constrained to the positive range and the
# interval (0, 1), respectively, we work on a transformed parameter space
# with unbounded domain.
def transform_parameters(params):
k, p = params
return np.exp(k), 1/(1+np.exp(-p))
# Log-Likelihood function for a negative binomial model
def logL(params):
k, p = transform_parameters(params)
return stats.nbinom.logpmf(data, k, p).sum()
# negative log-Likelihood function for optimization (because we use
# minimization algorithms instead of maximization algorithms)
negLogL = lambda params: -logL(params)
# Initial guess
x0 = [0, 0]
# Maximize the likelihood
result = op.minimize(negLogL, x0)
# Print the result (we need to transform the parameters to the original
# parameter space to make them interpretable)
print("The estimate is: k={:5.3f}, p={:5.3f}".format(*transform_parameters(result.x)))
# Find 95% confidence intervals for all parameters.
# Note: For complicated problems, it is worthwhile to do this in parallel.
# However, then we would need to encapsulate the procedure in a
# method and define the likelihood function on the top level of the module.
CIs = find_CI(result.x, logL, alpha=0.95,
disp=True) # the disp argument lets the algorithm print
# status messages.
# CIs is a 2D numpy array with CIs[i, 0] containing the lower bound of the
# confidence interval for the i-th parameter and CIs[i, 1] containing the
# respective upper bound.
# Print the confidence intervals. Note: we need to transform the parameters
# back to the original parameter space.
original_lower = transform_parameters(CIs[:,0])
original_upper = transform_parameters(CIs[:,1])
print("Confidence interval for k: [{:5.3f}, {:5.3f}]".format(
original_lower[0], original_upper[0]))
print("Confidence interval for n: [{:5.3f}, {:5.3f}]".format(
original_lower[1], original_upper[1]))
# Now we find the confidence interval for a function of the parameters,
# such as the sum of them
def myFunction(params):
return np.sum(transform_parameters(params))
CI = find_function_CI(result.x, myFunction, logL, alpha=0.95,
disp=True)
print(CI)
# Computing the gradient and Hessian is necessary for the algorithm
# but can be computationally expensive. If you want to have full
# control over this, define them manually:
# import a package to compute gradient and Hessian numerically
import numdifftools as nd
# Define gradient and Hessian
jac = nd.Gradient(logL)
hess = nd.Hessian(logL)
CIs = find_CI(result.x, logL, jac=jac, hess=hess, alpha=0.95,
disp=True) # the disp argument lets the algorithm print
# status messages.
Usage from R#
The package can also be used from R, if the package reticulate is available. You may install this package via the following command:
install.packages("reticulate")
Load the package.
library(reticulate)
Optional: If you want to use a python installation that already exists on your system, make sure it is in the PATH environment variable so that it can be found by R. Otherwise, specify your python executable via
use_python("/my/path/to/python")
Optional: If you use python via Anaconda, you can activate your environment of preference via the command below. If you are not sure what this does, you are presumably fine and can omit this line.
use_virtualenv("base")
Here, “base” is the name of the environment.
If the package with the algorithm is not installed on your system yet, run
py_install("ci-rvm", pip=TRUE)
If python is not yet installed on your system, you will be asked to install Miniconda when loading the package. We advise you to do this.
Now everything should be set up for using the algorithm. Below you can find an example R script.
# ========== imports ===========
# We start by importing the library that builds a bridge to python.
library(reticulate)
# Now we import the python package to compute profile likelihood confidence
# intervals.
ci_rvm = import("ci_rvm")
# ========== loading data and defining likelihood function ===========
# We consider the built-in data on height an weight of women. We assume that the
# weight is normally distributed around a polynomial function of the height:
# meanWeight = a + b * height^c
# where a, b, and c are parameters.
# We assume that the variance is proportional to the mean height:
# varianceWeight = d * meanWeight
# The log-likelihood function looks as follows (up to a constant)
logLikelihood = function(parameters) {
a = abs(parameters[1]) # constrain this parameter to the positive range
b = parameters[2]
c = abs(parameters[3]) # constrain this parameter to the positive range
meanWeight = a * women["height"]^b
varianceWeight = c * meanWeight
result = (-sum(log(varianceWeight))/2
-sum((women["weight"]-meanWeight)^2/(2*varianceWeight)))
return(result)
}
# ========== defining gradient and Hessian of the log-likelihood ===========
# Maximizing the likelihood and finding confidence intervals requires knowledge
# of the gradient (vector of first derivatives) and Hessian
# (matrix of second derivatives) of the log-likelihood function.
# For the considered example, we could compute these by hand on paper. However,
# to show how we could proceed in more complicated cases, we use packages to
# compute the derivatives.
# If we do not provide gradient and Hessian, the package will compute them
# for us, but this could potentially be less efficient when working from R.
install.packages("numDeriv") # omit this line, if the library is already installed
library("numDeriv")
gradientLL = function(parameters) {
return(grad(logLikelihood, parameters))
}
hessianLL = function(parameters) {
return(hessian(logLikelihood, parameters))
}
# ========== maximizing the likelihood ===========
# define initial guess
guess = c(1, 1, 1)
# maximize the likelihood
result = optim(guess, logLikelihood, gr=gradientLL,
control=list(fnscale=-1, maxit=3000, trace=0))
# estimated parameters
estimate = result$par
# check the result by plotting estimated mean and data
height = (min(women["height"])-5):(max(women["height"])+5)
weight = abs(estimate[1]) * height^estimate[2]
plot(height, weight, type="l")
points(unlist(women["height"], use.names=FALSE), unlist(women["weight"], use.names=FALSE))
# ========== computing profile likelihood confidence intervals ===========
# Compute all confidence intervals at once. We obtain a matrix with a row for
# each parameter and the lower and upper confidence interval limit as columns.
# If return_success=TRUE, a second matrix will be returned, indicating which
# of the bounds have been found with success.
# Set disp=TRUE to print status information. Especially if something goes wrong
# (which could, e.g., happen if the provided estimate is not close to the actual
# maximum), this information can be very useful.
# alpha denotes the desired confidence level (0.95 by default)
confidenceIntervals = ci_rvm$find_CI(estimate, logLikelihood, gradientLL,
hessianLL, alpha=0.95, disp=TRUE,
return_success=FALSE)
print(confidenceIntervals)
# Note that automatic parallel search for multiple confidence interval bounds
# does not work if the code is called from R. You will have to do the
# parallelization manually from within R. To that end, you can determine an
# individual confidence interval bound as follows.
# Set the index of the parameter under investigation. Note that python indices
# start at 0.
index = 0
# Set the search direction. Use a positive value or TRUE to find the upper
# end point of the confidence interval and a negative value or FALSE to find
# the lower end point.
direction = -1
# Now execute the search.
confidenceInterval1Lower = ci_rvm$find_CI(estimate, logLikelihood,
gradientLL, hessianLL, index, direction, alpha=0.95, disp=FALSE,
track_x=TRUE, # If we want to track what the algorithm did
return_full_results=TRUE # If we want to access details on the result
)
# The resulting object also contains information on what the other parameters
# were when the parameter under consideration assumed its extreme value.
# This information can be helpful to detect connections between parameters.
print(confidenceInterval1Lower$x)
# ------- additional fun stuff ---------
# Similarly, we could plot the trajectory of the search, This requires that
# track_x=TRUE when searching the confidence interval.
# First, we plot the relationship between the first two parameters.
plot(confidenceInterval1Lower$x_track[,1], confidenceInterval1Lower$x_track[,2],
type="o", xlab="Weight factor", ylab="Weight exponent")
# The line above seems to be super straight and we may wonder why so many
# steps were needed to find the confidence interval end point.
# Let us consider the relationship between the first and the third parameter.
plot(confidenceInterval1Lower$x_track[,1], confidenceInterval1Lower$x_track[,3],
type="o", xlab="Weight factor", ylab="Variance factor")
# We see the search trajectory is curved - a sign that the likelihood surface
# is not trivial.
# When we compute the Wald confidence intervals for comparison, we see that
# they are good for the first two parameters, but not as good for the last
# one:
deviation = sqrt(-diag(solve(hessianLL(estimate))) * qchisq(0.95, df = 1))
waldCI = cbind(estimate - deviation, estimate + deviation)
print("Wald CI")
print(waldCI)
print("Profile CI")
print(confidenceIntervals)
print("Relative Error")
print(1-waldCI/confidenceIntervals)
# ------------
# We can also compute the confidence interval for a function of parameters,
# such as the mean expected value of the distribution
meanExpectedValue = function(parameters) {
a = abs(parameters[1]) # constrain this parameter to the positive range
b = parameters[2]
meanWeight = a * women["height"]^b
return(sum(meanWeight))
}
gradientMean = function(parameters) {
return(grad(meanExpectedValue, parameters))
}
hessianMean = function(parameters) {
return(hessian(meanExpectedValue, parameters))
}
ciOfMean = ci_rvm$find_function_CI(estimate, meanExpectedValue, logLikelihood,
gradientMean, hessianMean, gradientLL,
hessianLL, alpha=0.95, disp=FALSE
)
print(ciOfMean)
Scientific Publication#
The theory behind the algorithm implemented in this package is explained in the paper “A robust and efficient algorithm to find profile likelihood confidence intervals”. Please cite this publication if you have used the package in your own research.